22 lines
437 B
Python
22 lines
437 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 l.startswith("counts_sum=")
|
|
]
|