fixed test

romanshupak · Sep 9, 2022 · ee2952f · ee2952f
1 parent 5997554
commit ee2952f
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -1,4 +1,5 @@
 import pytest
+import re
 
 from app.main import count_occurrences
 
@@ -22,10 +23,10 @@ def test_count_occurrences(phrase, letter, count):
 
 
 def test_removed_comment():
-import app
-with open(app.main.__file__, "r") as f:
-    file_content = f.read()
-    comment = re.compile("# write your code here")
-    assert not comment.search(
-        file_content
-    ), "You have to remove the unnecessary comment '# write your code here'"
+    import app
+    with open(app.main.__file__, "r") as f:
+        file_content = f.read()
+        comment = re.compile("# write your code here")
+        assert not comment.search(
+            file_content
+        ), "You have to remove the unnecessary comment '# write your code here'"