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_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)
def test_gives_none_if_no_taken_week_specified(self):
y = Year(2012)
self.assertEquals(y.get_taken(40), None)
