class CatFinderFunctions(TestCase): def setUp(self): self.cat_finder = CatFinder(0) def test_initialise_cats(self): self.cat_finder.initialise_cats(2) self.assertEquals(self.cat_finder.lost_cats[1].number, 1)
class CatFinderFoundCatTest(TestCase): def setUp(self): self.cat_finder = CatFinder(5) self.cat_finder.cat_found(self.cat_finder.lost_cats[0]) def test_station_closed(self): self.assertFalse(self.cat_finder.found_cats[0].current_station.opened) def test_number_of_lost_cats(self): self.assertEqual(len(self.cat_finder.lost_cats), 4) def test_number_of_found_cats(self): self.assertEqual(len(self.cat_finder.found_cats), 1)
def setUp(self): self.cat_finder = CatFinder(0)
def setUp(self): self.cat_finder = CatFinder(5) self.cat_finder.cat_found(self.cat_finder.lost_cats[0])