Merge pull request #178 from mate-academy/add-inspect

added inspect

Author: Nattalli

tests/test_main.py

@@ -1,3 +1,5 @@
+import inspect
+
 import pytest
 
 from app.main import count_occurrences
@@ -12,19 +14,17 @@
         ("Abracadabra", "A", 5),
         ("", "a", 0),
         ("Samsung", "b", 0),
-    ]
+    ],
 )
 def test_count_occurrences(phrase, letter, count):
     assert count_occurrences(phrase, letter) == count, (
         f"Function 'count_occurrences' should return {count}, "
         f"when 'phrase'='{phrase}' and 'letter'='{letter}'"
     )
 
-    
-def test_removed_comment():
-    import app
-    with open(app.main.__file__, "r") as f:
-        file_content = f.read()
-        assert "# write your code here" not in file_content, \
-            "You have to remove the unnecessary comment '# write your code here'"
 
+def test_removed_comment():
+    lines = inspect.getsource(count_occurrences)
+    assert "# write your code here" not in lines, (
+        "You have to" " remove the unnecessary comment '# write your code here'"
+    )