jfz-2023-s473555/TaskD01/run.py

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")