class TestCandidateVotes(TestCase): def setUp(self): self.cand = Candidate("Smith, Joe", "GOP") def test_default_zero_votes(self): "Candidate vote count should default to zero" self.assertEquals(self.cand.votes, 0) def test_vote_count_update(self): "Candidate.add_votes method should update vote count" self.cand.add_votes("Some County", 20) self.assertEquals(self.cand.votes, 20) def test_county_results_access(self): "Candidate.add_votes method should store county results" self.cand.add_votes("Some County", 20) expected = {"Some County": 20} self.assertEquals(self.cand.county_results, expected)
class TestCandidateVotes(TestCase): def setUp(self): self.cand = Candidate("Smith, Joe", "GOP") def test_default_zero_votes(self): "Candidate vote count should default to zero" self.assertEquals(self.cand.votes, 0) def test_vote_count_update(self): "Candidate.add_votes method should update vote count" self.cand.add_votes("Some County", 20) self.assertEquals(self.cand.votes, 20) def test_county_results_access(self): "Candidate.add_votes method should store county results" self.cand.add_votes("Some County", 20) expected = { "Some County": 20 } self.assertEquals(self.cand.county_results, expected)
def test_candidate_name(self): "Candidates should have first_name and last_name attributes" cand = Candidate("Smith, Joe", "GOP") self.assertEquals(cand.first_name, "Joe") self.assertEquals(cand.last_name, "Smith")
def setUp(self): self.cand = Candidate("Smith, Joe", "GOP")
def test_clean_party_others(self): cand = Candidate("Smith, Joe", "Green") self.assertEquals(cand.party, 'GREEN')
def test_clean_party_dem(self): cand = Candidate("Smith, Joe", "Democratic") self.assertEquals(cand.party, 'DEM')
def test_party_gop(self): cand = Candidate("Smith, Joe", "GOP") self.assertEquals(cand.party, 'REP')