Update test_main.py

romanshupak · Sep 9, 2022 · 858ac8b · 858ac8b
1 parent ee2952f
commit 858ac8b
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -1,5 +1,4 @@
 import pytest
-import re
 
 from app.main import count_occurrences
 
@@ -26,7 +25,6 @@ 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'"
+        assert "# write your code here" not in file_content, \
+            "You have to remove the unnecessary comment '# write your code here'"
+