def test_can_retrieve_taken_week(self): y = Year(2012) taken = Week() y.add_taken(40, taken) self.assertEquals(y.get_taken(40), taken)
def test_gives_none_if_no_taken_week_specified(self): y = Year(2012) self.assertEquals(y.get_taken(40), None)