mirror of
https://github.com/coding-horror/basic-computer-games.git
synced 2025-12-22 07:10:42 -08:00
31 lines
952 B
Python
31 lines
952 B
Python
import io
|
|
|
|
import hamurabi
|
|
|
|
|
|
def test_main(monkeypatch, capsys):
|
|
monkeypatch.setattr("sys.stdin", io.StringIO("100\n100\n100"))
|
|
hamurabi.main()
|
|
captured = capsys.readouterr()
|
|
actual_lines = captured.out.splitlines()
|
|
expected_lines = [
|
|
"HAMURABI", # 0
|
|
"CREATIVE COMPUTING MORRISTOWN, NEW JERSEY", # 1
|
|
"", # 2
|
|
"", # 3
|
|
"", # 4
|
|
"", # 5
|
|
"TRY YOUR HAND AT GOVERNING ANCIENT SUMERIA", # 6
|
|
"FOR A TEN-YEAR TERM OF OFFICE.", # 7
|
|
"", # 8
|
|
"", # 9
|
|
"", # 10
|
|
"", # 11
|
|
"HAMURABI: I BEG TO REPORT TO YOU\n", # 12
|
|
"IN YEAR 1 , 0 PEOPLE STARVED, 5 CAME TO THE CITY,\n", # 13
|
|
"POPULATION IS NOW 100\n", # 14
|
|
"THE CITY NOW OWNS 1000.0 ACRES.", # 15
|
|
]
|
|
for i, (actual, expected) in enumerate(zip(actual_lines, expected_lines)):
|
|
assert actual.strip() == expected.strip(), f"Line {i} is wrong"
|