fix integration test

pixee · Mar 2, 2024 · bdcb4e8 · bdcb4e8
1 parent 3fe9453
commit bdcb4e8
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/integration_tests/test_sonar_exception_without_raise.py b/integration_tests/test_sonar_exception_without_raise.py
@@ -1,15 +1,13 @@
-from core_codemods.exception_without_raise import (
-    ExceptionWithoutRaise,
-    ExceptionWithoutRaiseTransformer,
-)
+from core_codemods.exception_without_raise import ExceptionWithoutRaiseTransformer
+from core_codemods.sonar.sonar_exception_without_raise import SonarExceptionWithoutRaise
 from codemodder.codemods.test import (
     BaseIntegrationTest,
     original_and_expected_from_code_path,
 )
 
 
 class TestSonarExceptionWithoutRaise(BaseIntegrationTest):
-    codemod = ExceptionWithoutRaise
+    codemod = SonarExceptionWithoutRaise
     code_path = "tests/samples/exception_without_raise.py"
     original_code, expected_new_code = original_and_expected_from_code_path(
         code_path,