def test_popular_vote_2_riding_2_party() -> None:
"""Test popular_vote for 2 riding and 2 party"""
e = Election(date(2000, 2, 8))
e.update_results('r1', 'ndp', 19)
e.update_results('r2', 'np', 69)
res1 = e.popular_vote()
assert res1 == {'ndp': 19, 'np': 69}
def test_popular_vote_0_votes() -> None:
"""Test popular_vote for each party has 0 votes"""
e = Election(date(2000, 2, 8))
e.update_results('r1', 'ndp', 0)
e.update_results('r2', 'np', 0)
res1 = e.popular_vote()
assert res1 == {'ndp': 0, 'np': 0}
def test_complex_election_popular_vote() -> None:
"""Test Election.popular_vote with a simple Election."""
e = Election(date(2000, 2, 8))
e.update_results('r1', 'ndp', 1)
e.update_results('r1', 'lib', 1)
e.update_results('r1', 'pc', 0)
e.update_results('r2', 'ndp', 3)
e.update_results('r2', 'lib', 2)
e.update_results('r2', 'pc', 0)
e.update_results('r2', 'green', 1)
assert e.popular_vote() == {'ndp': 4, 'lib': 3, 'green': 1, 'pc': 0}
def test_one_party_one_riding_read_results() -> None:
"""Test Election.read_results with a file with a single line."""
file = StringIO(SHORT_FILE_CONTENTS)
e = Election(date(2012, 10, 30))
e.read_results(file)
assert e.popular_vote() == {'Liberal': 113}
def test_popular_vote_1_riding_1_party() -> None:
"""Test popular_vote for 1 riding and 1 party"""
e = Election(date(2000, 2, 8))
e.update_results('r1', 'ndp', 19)
res1 = e.popular_vote()
assert res1 == {'ndp': 19}