def test_retrieve_donor_exists():
mailroom.DONORS = {'Fred Armisen': [240, 422, 1000],
'Heinz the Baron Krauss von Espy': [1500, 2300],
'Margaret Atwood': [300, 555]}
name = "Margaret Atwood"
output = [300, 555]
assert mailroom.retrieve_donations(name) == output
def test_add_donation_existing():
mailroom.DONORS = {'Fred Armisen': [240, 422, 1000],
'Heinz the Baron Krauss von Espy': [1500, 2300],
'Margaret Atwood': [300, 555]}
name = "Fred Armisen"
amount = 500
mailroom.add_donation(name, amount)
assert mailroom.retrieve_donations(name)[-1] == amount
def test_add_donation_new():
mailroom.DONORS = {'Fred Armisen': [240, 422, 1000],
'Heinz the Baron Krauss von Espy': [1500, 2300],
'Margaret Atwood': [300, 555]}
name = "Bob Mueller"
amount = 500
mailroom.add_donation(name, amount)
assert mailroom.retrieve_donations(name) == [amount]
def test_retreive_donations():
result = mailroom.retrieve_donations('John Smith')
print(result)
assert result == [25,25]
def test_notget_donations():
result = mailroom.retrieve_donations('Mathew Martin')
assert result is None
def test_retrieve_donations_new():
mailroom.DONORS = {'Fred Armisen': [240, 422, 1000],
'Heinz the Baron Krauss von Espy': [1500, 2300],
'Margaret Atwood': [300, 555]}
assert mailroom.retrieve_donations("Paul Simon") is None