24 lines
458 B
Python
24 lines
458 B
Python
import sys
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from solution import main
|
|
|
|
|
|
HERE = Path(__file__).absolute().parent
|
|
|
|
|
|
def test_solution(capsys):
|
|
with (HERE / "test-input").open() as infile:
|
|
sys.stdin = infile
|
|
main()
|
|
|
|
expected_output = (HERE / "test-output").read_text().splitlines()
|
|
assert expected_output == [
|
|
l
|
|
for l in capsys.readouterr().out.splitlines()
|
|
if not l.startswith("highest_seat")
|
|
]
|