Update test_main.py

Author: Nattalli

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'"
+