Added test with __init__.py file for RemoveUnusedImports

pixee · Oct 26, 2023 · 39edc0f · 39edc0f
1 parent cd6a305
commit 39edc0f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/codemods/test_remove_unused_imports.py b/tests/codemods/test_remove_unused_imports.py
@@ -1,3 +1,4 @@
+from pathlib import Path
 from core_codemods.remove_unused_imports import RemoveUnusedImports
 from tests.codemods.base_codemod_test import BaseCodemodTest
 from textwrap import dedent
@@ -113,3 +114,9 @@ def test_dont_remove_if_pylint_disable_next(self, tmpdir):
         )
         self.run_and_assert(tmpdir, before, before)
         assert len(self.file_context.codemod_changes) == 0
+
+    def test_ignore_init_files(self, tmpdir):
+        before = "import a"
+        tmp_file_path = Path(tmpdir / "__init__.py")
+        self.run_and_assert_filepath(tmpdir, tmp_file_path, before, before)
+        assert len(self.file_context.codemod_changes) == 0