fix test

tests/test_identical.py

@@ -8,17 +8,19 @@
 
 
 def test_identical(solution_file: Path):
-    level, challenge_name = solution_file.parent.name.split("-", maxsplit=1)
-    with solution_file.open() as f:
-        solution_code = f.read()
-    solution_test = Challenge(
-        name=challenge_name, level=Level(level), code=solution_code
-    ).test_code
-
-    question_file = solution_file.parent / "question.py"
-    with question_file.open() as f:
-        question_code = f.read()
-    question_test = Challenge(
-        name=challenge_name, level=Level(level), code=question_code
-    ).test_code
+    def get_test_code(path: Path):
+        TEST_SPLITTER = "\n## End of test code ##\n"
+        level, challenge_name = path.parent.name.split("-", maxsplit=1)
+
+        with solution_file.open() as f:
+            challenge_code = f.read()
+        challenge = Challenge(
+            name=challenge_name, level=Level(level), code=challenge_code
+        )
+
+        return challenge.test_code.partition(TEST_SPLITTER)[0]
+
+    solution_test = get_test_code(solution_file)
+    question_test = get_test_code(solution_file.parent / "question.py")
+
     assert solution_test.strip() == question_test.strip()