38 lines
868 B
Python
38 lines
868 B
Python
|
import re
|
||
|
|
|
||
|
|
file_name: str = "simple"
|
||
|
|
|
||
|
|
|
||
|
|
def find_hamlet_lines(text: str) -> list[str]:
|
||
|
|
# Define the regular expression pattern
|
||
|
|
pattern = re.compile(r".*Hamlet.*", re.IGNORECASE)
|
||
|
|
|
||
|
|
# Split the text into lines
|
||
|
|
lines = text.split("\n")
|
||
|
|
|
||
|
|
# Use the regular expression to find lines containing "Hamlet"
|
||
|
|
hamlet_lines = [line for line in lines if re.match(pattern, line)]
|
||
|
|
|
||
|
|
return hamlet_lines
|
||
|
|
|
||
|
|
|
||
|
|
with open(file_name + ".in", "r", newline="", encoding="utf8") as file:
|
||
|
|
text = file.read()
|
||
|
|
|
||
|
|
with open(file_name + ".exp", "r", newline="", encoding="utf8") as file:
|
||
|
|
expected = [line.rstrip() for line in file.readlines()]
|
||
|
|
|
||
|
|
|
||
|
|
found_lines = find_hamlet_lines(text)
|
||
|
|
|
||
|
|
mistake = False
|
||
|
|
for line in expected:
|
||
|
|
if line in found_lines:
|
||
|
|
print(f"{line}")
|
||
|
|
else:
|
||
|
|
print(f"NO: {line}")
|
||
|
|
mistake = True
|
||
|
|
|
||
|
|
if mistake:
|
||
|
|
print("ERROR")
|