import re file_name: str = "simple" def find_hamlet_lines(text: str) -> list[str]: pattern = re.compile(r"Hamlet", re.IGNORECASE) lines = text.split("\n") hamlet_lines = [line for line in lines if re.search(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")