From 73f5b4a8161daae04e49c1f9579eb5579df01ee7 Mon Sep 17 00:00:00 2001
From: andrecs <12188364+andrecsilva@users.noreply.github.com>
Date: Thu, 26 Oct 2023 11:01:31 -0300
Subject: [PATCH] Added test for exceptions in RemoveUnusedImports

---
 tests/codemods/test_remove_unused_imports.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tests/codemods/test_remove_unused_imports.py b/tests/codemods/test_remove_unused_imports.py
index 4d4a613d..9e675b5c 100644
--- a/tests/codemods/test_remove_unused_imports.py
+++ b/tests/codemods/test_remove_unused_imports.py
@@ -120,3 +120,15 @@ def test_ignore_init_files(self, tmpdir):
         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
+
+    def test_no_pyling_pragma_in_comment_trailing(self, tmpdir):
+        before = "import a # bogus: no-pragma"
+        after = ""
+        self.run_and_assert(tmpdir, before, after)
+        assert len(self.file_context.codemod_changes) == 1
+
+    def test_no_pyling_pragma_in_comment_before(self, tmpdir):
+        before = "#header\nprint('hello')\n# bogus: no-pragma\nimport a "
+        after = "#header\nprint('hello')"
+        self.run_and_assert(tmpdir, before, after)
+        assert len(self.file_context.codemod_changes) == 1