Tirsvad
7459711243
Closes #11 Closes #12 Closes #13 Closes #14 Closes #15 Task: MIL-002#4 Task: MIL-002#5 Task: MIL-002#6 Task: MIL-002#7 Task: MIL-002#8
22 lines
603 B
Python
22 lines
603 B
Python
"""Check the counts of the real 2018 Central Park census file, when present."""
|
|
|
|
import pathlib
|
|
|
|
import pytest
|
|
|
|
from squirrel import census
|
|
from squirrel.constants import DEFAULT_INPUT_PATH, FUR_COLORS
|
|
|
|
REPO_ROOT = pathlib.Path(__file__).parents[1]
|
|
|
|
|
|
def test_real_census_counts_match_published_figures() -> None:
|
|
path = REPO_ROOT / DEFAULT_INPUT_PATH
|
|
if not path.is_file():
|
|
pytest.skip(f"census file not present: {path}")
|
|
|
|
counts = census.count_by_fur_color(census.read_csv(path))
|
|
|
|
gray, cinnamon, black = FUR_COLORS
|
|
assert counts == {gray: 2473, cinnamon: 392, black: 103}
|