def test_init_add_garage_instance(self): cesar = Cesar(rnd_name()) garage = Garage(rnd_town(), rnd_int()) for value in [1000, 'some string', [garage]]: with self.assertRaises(TypeError): cesar.add_garage(value)
def test_init_add_garage(self): garage1 = Garage(rnd_town(), rnd_int()) garage2 = Garage(rnd_town(), rnd_int()) cesar = Cesar(rnd_name(), [garage1]) self.assertFalse(cesar.add_garage(garage1)) self.assertTrue(cesar.add_garage(garage2)) self.assertEqual(cesar.register_id, garage1.owner) self.assertEqual(cesar.register_id, garage2.owner) self.assertTrue(garage1.number in cesar.garages) self.assertTrue(garage2.number in cesar.garages)