import sys

import tempfile

from collections.abc import Callable

from pathlib import Path

def loop_inputs(pattern: str, function: Callable) -> None:

files = Path(__file__).parent.joinpath("data").glob(f"{pattern}.input")

for input_file in files:

output_file = input_file.with_suffix(".output")

with (

Path(input_file).open(encoding="utf-8") as sys.stdin,

tempfile.NamedTemporaryFile(

mode="w+t", encoding="utf-8", delete=True, delete_on_close=True,

) as temp_file,

):

function(temp_file.file)

temp_file.file.seek(0)

with Path(output_file).open(mode="r+t", encoding="utf-8") as output_file:

assert (

output_file.read() == temp_file.read()

), f"Answer failed for {Path(input_file).name}"

output_file.close()

temp_file.close()