def test_get_donor(): # Test that donors can be retrieved from collection dc = DonorCollection() dc.updateDonor('Jason Mendoza', 100.) d = dc.getDonor('Jason Mendoza') assert isinstance(d, Donor) assert d.name == 'Jason Mendoza' assert d.lastDonation() == 100. # Should return new donor if donor does not exist d = dc.getDonor('New Donor') assert isinstance(d, Donor) assert d.name == 'New Donor'
def test_update_donor(): # Tests that new donor can be added and updated dc = DonorCollection() dc.updateDonor('Jason Mendoza', 100.) dc.updateDonor('Jason Mendoza', 200.) d = dc.getDonor('Jason Mendoza') assert d.lastDonation() == 200. assert d.totalDonations() == 300. assert d.numDonations() == 2 assert d.averageDonation() == 150.