def test_party_wins_1_election_lose() -> None: """Test party_wins for 1 election and the party losing.""" e1 = Election(date(2000, 2, 8)) e1.update_results('r1', 'ndp', 1) e1.update_results('r1', 'lib', 2) e1.update_results('r1', 'pc', 3) e1.update_results('r2', 'lib', 10) e1.update_results('r2', 'pc', 20) e1.update_results('r3', 'ndp', 200) e1.update_results('r3', 'pc', 100) j = Jurisdiction('Canada') j._history[date(2000, 2, 8)] = e1 res1 = j.party_wins('lib') assert res1 == []
def test_party_wins_2_election_all_tie() -> None: """Test party_wins for 1 election and the party losing.""" e1 = Election(date(2000, 2, 8)) e1.update_results('r1', 'ndp', 10) e1.update_results('r1', 'lib', 100) e1.update_results('r1', 'pc', 100) e1.update_results('r2', 'lib', 10) e1.update_results('r2', 'pc', 20) e1.update_results('r3', 'ndp', 200) e1.update_results('r3', 'pc', 100) e2 = Election(date(2004, 5, 16)) e2.update_results('r1', 'ndp', 10) e2.update_results('r1', 'lib', 2) e2.update_results('r2', 'lib', 30) e2.update_results('r2', 'ndp', 5) j = Jurisdiction('Canada') j._history[date(2000, 2, 8)] = e1 j._history[date(2004, 5, 16)] = e2 res1 = j.party_wins('ndp') assert res1 == [date(2000, 2, 8), date(2004, 5, 16)]