def test_legislator_related_district(): leg = Person("John Adams", district="1", primary_org="legislature") leg.pre_save("jurisdiction-id") assert len(leg._related) == 1 assert leg._related[0].person_id == leg._id assert get_pseudo_id(leg._related[0].organization_id) == { "classification": "legislature" } assert get_pseudo_id(leg._related[0].post_id) == { "organization__classification": "legislature", "label": "1", }
def test_simple_vote_event(): ve = toy_vote_event() ve.set_count("yes", 2) ve.yes("James") ve.no("Paul") ve.vote("abstain", "Thom") assert len(ve.votes) == 3 assert len(ve.counts) == 1 assert get_pseudo_id(ve.organization) == {"classification": "legislature"} assert get_pseudo_id(ve.votes[0]["voter_id"]) == {"name": "James"} assert get_pseudo_id(ve.votes[1]["voter_id"]) == {"name": "Paul"} assert get_pseudo_id(ve.votes[2]["voter_id"]) == {"name": "Thom"} assert ve.bill is None ve.validate() assert "we get here"
def test_set_bill_pseudo_id(): ve = toy_vote_event() ve.set_bill("HB 1", chamber="lower") assert get_pseudo_id(ve.bill) == { "identifier": "HB 1", "from_organization__classification": "lower", "legislative_session__identifier": "2009", }
def test_person_add_party(): p = Person("Groot") p.add_party("Green") p._related[0].validate() assert get_pseudo_id(p._related[0].organization_id) == { "name": "Green", "classification": "party", }
def test_person_add_term(): p = Person("Eternal") p.add_term("eternal", "council", start_date="0001", end_date="9999") p._related[0].validate() assert get_pseudo_id(p._related[0].organization_id) == { "classification": "council" } assert p._related[0].start_date == "0001" assert p._related[0].end_date == "9999"
def test_legislator_related_chamber_district_role(): leg = Person("John Adams", district="1", primary_org="lower", role="Speaker") leg.pre_save("jurisdiction-id") assert len(leg._related) == 1 assert leg._related[0].person_id == leg._id assert get_pseudo_id(leg._related[0].organization_id) == { "classification": "lower" } assert get_pseudo_id(leg._related[0].post_id) == { "organization__classification": "lower", "label": "1", "role": "Speaker", } assert leg._related[0].role == "Speaker"
def test_vote_event_org_chamber(): ve = VoteEvent( legislative_session="2009", motion_text="passage of the bill", start_date="2009-01-07", result="pass", classification="passage", chamber="upper", ) assert get_pseudo_id(ve.organization) == {"classification": "upper"}
def test_vote_event_org_dict(): odict = {"name": "Random Committee", "classification": "committee"} ve = VoteEvent( legislative_session="2009", motion_text="passage of the bill", start_date="2009-01-07", result="pass", classification="passage", organization=odict, ) assert get_pseudo_id(ve.organization) == odict
def test_person_add_membership_name(): p = Person("Leonardo DiCaprio") p.add_membership("Academy of Motion Picture Arts and Sciences", role="winner", start_date="2016") p._related[0].validate() assert get_pseudo_id(p._related[0].organization_id) == { "name": "Academy of Motion Picture Arts and Sciences" } assert p._related[0].person_id == p._id assert p._related[0].role == "winner" assert p._related[0].start_date == "2016"
def test_legislator_related_party(): leg = Person("John Adams", party="Democratic-Republican") leg.pre_save("jurisdiction-id") # a party membership assert len(leg._related) == 1 assert leg._related[0].person_id == leg._id assert get_pseudo_id(leg._related[0].organization_id) == { "classification": "party", "name": "Democratic-Republican", } assert leg._related[0].role == "member"
def test_committee_add_member_name(): c = Organization("Defense", classification="committee") c.add_member("John Adams") assert get_pseudo_id(c._related[0].person_id) == {"name": "John Adams"} assert c._related[0].organization_id == c._id assert c._related[0].role == "member"