def test_expected_districts_vacancies(): expected = get_expected_districts({ "upper_seats": 3, "lower_seats": { "At-Large": 3 }, "vacancies": [ { "chamber": "upper", "district": "2", "vacant_until": datetime.date(2100, 1, 1) }, { "chamber": "lower", "district": "At-Large", "vacant_until": datetime.date(2100, 1, 1) }, ] }) assert expected['upper'] == {"1": 1, "2": 0, "3": 1} assert expected['lower'] == {"At-Large": 2} with pytest.raises(BadVacancy): get_expected_districts({ "upper_seats": 3, "vacancies": [ { "chamber": "upper", "district": "2", "vacant_until": datetime.date(2000, 1, 1) }, ] })
def test_expected_districts_vacancies(): vacancies = { "ne": { "vacancies": [{ "chamber": "legislature", "district": "1", "vacant_until": datetime.date(2100, 1, 1), }] } } expected = get_expected_districts(vacancies, "ne") assert expected["legislature"]["1"] == 0 with pytest.raises(BadVacancy): get_expected_districts( { "ne": { "vacancies": [{ "chamber": "upper", "district": "2", "vacant_until": datetime.date(2000, 1, 1), }] } }, "ne", )
def test_get_expected_districts(): expected = get_expected_districts({}, "ne") assert len(expected["legislature"]) == 49 assert expected["legislature"]["1"] == 1 expected = get_expected_districts({}, "md") print(expected) assert expected["lower"]["3A"] == 2 assert expected["lower"]["3B"] == 1
def test_get_expected_districts(): expected = get_expected_districts({"upper_seats": 3, "lower_seats": ["A", "B", "C"], "legislature_seats": {"At-Large": 3}}) assert expected['upper'] == {"1": 1, "2": 1, "3": 1} assert expected['lower'] == {"A": 1, "B": 1, "C": 1} assert expected['legislature'] == {"At-Large": 3}