def test_init_add_car(self):
garage1 = Garage(rnd_town(), 1)
garage2 = Garage(rnd_town(), 2)
cesar = Cesar(rnd_name(), [garage1])
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car3 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
cesar.add_car(car1, garage1)
cesar.add_car(car2, garage2)
cesar.add_car(car3, garage2)
self.assertEqual(cesar.register_id, garage1.owner)
self.assertEqual(cesar.register_id, garage2.owner)
self.assertEqual(garage1.cars_count(), 1)
self.assertEqual(garage2.cars_count(), 2)
self.assertTrue(car1.number in garage1.cars)
self.assertTrue(car2.number in garage2.cars)
self.assertTrue(car3.number in garage2.cars)
def test_places_count(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), 5, rnd_uuid(), [car1, car2])
self.assertEqual(garage.places_count(), 3)
def test_comparison(self):
car1 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car2 = Car(2000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car3 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
self.assertGreater(car2, car1)
self.assertGreaterEqual(car1, car3)
self.assertLess(car1, car2)
self.assertLessEqual(car1, car3)
def test_hit_hat(self):
car1 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car2 = Car(2000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car3 = Car(3000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1])
garage2 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car2, car3])
cesar = Cesar(rnd_name(), [garage1, garage2])
self.assertEqual(cesar.hit_hat(), 6000)
def test_not_equal_price(self):
price = rnd_float()
car_type = rnd_car_type()
car_producer = rnd_car_producer()
mileage = rnd_float()
number = rnd_uuid()
car1 = Car(price, car_type, car_producer, mileage, number)
car2 = Car(price + 1, car_type, car_producer, mileage, number)
self.assertNotEqual(car1, car2)
def test_hit_hat(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1, car2])
self.assertEqual(garage.hit_hat(), car1.price + car2.price)
garage = Garage(rnd_town(), rnd_int(), rnd_uuid())
self.assertEqual(garage.hit_hat(), 0)
def test_add_car(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), 10, rnd_uuid(), [car1])
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
self.assertTrue(garage.add(car2))
self.assertTrue(car2.number in garage.cars)
self.assertEqual(garage.cars[car2.number], car2)
def test_remove_not_existing_car(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1])
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
before_cars = garage.cars
self.assertFalse(garage.remove(car2))
after_cars = garage.cars
self.assertEqual(before_cars, after_cars)
def test_add_when_no_places(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), 1, rnd_uuid(), [car1])
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
before_cars = garage.cars
self.assertFalse(garage.add(car2))
after_cars = garage.cars
self.assertEqual(before_cars, after_cars)
def test_cars_count(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car3 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1])
garage2 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car2, car3])
cesar = Cesar(rnd_name(), [garage1, garage2])
self.assertEqual(cesar.cars_count(), 3)
def test_json_serialize(self):
filename = 'car.json'
car_from_json = Car.from_json_file(filename)
with open(get_dump_path(filename)) as f:
car_json = json.loads(f.read())
car = Car(price=car_json.get('price'),
car_type=car_json.get('car_type'),
producer=car_json.get('producer'),
mileage=car_json.get('mileage'),
number=car_json.get('number'))
self.assertEqual(car, car_from_json)
def test_not_equal_number(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
town = rnd_town()
places = rnd_int()
owner = rnd_uuid()
cars = [car1, car2]
number = rnd_uuid()
garage1 = Garage(town, places, owner, cars, number)
garage2 = Garage(town, places, owner, cars, rnd_uuid())
self.assertNotEqual(garage1, garage2)
def test_yaml_serialize(self):
filename = 'cesar.yaml'
cesar_from_yaml = Cesar.from_yaml_file(filename)
with open(get_dump_path(filename)) as f:
cesar_json = yaml.load(f.read(), Loader=yaml.FullLoader)
garages = []
for garage_json in cesar_json.get('garages'):
cars = []
for car_json in garage_json.get('cars'):
car = Car(price=car_json.get('price'),
car_type=car_json.get('car_type'),
producer=car_json.get('producer'),
mileage=car_json.get('mileage'),
number=car_json.get('number'))
cars.append(car)
garage = Garage(town=garage_json.get('town'),
places=garage_json.get('places'),
owner=garage_json.get('owner'),
cars=cars,
number=garage_json.get('number'))
garages.append(garage)
cesar = Cesar(name=cesar_json.get('name'),
garages=garages,
register_id=cesar_json.get('register_id'))
self.assertEqual(cesar, cesar_from_yaml)
def test_init_add_car_instance(self):
cesar = Cesar(rnd_name())
car = Car(rnd_float(), rnd_car_type(), rnd_car_producer(), rnd_float())
for value in [1000, 'some string', [car]]:
with self.assertRaises(TypeError):
cesar.add_car(value)
def test_remove_car(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), 10, rnd_uuid(), [car1])
self.assertTrue(garage.remove(car1))
self.assertTrue(car1.number not in garage.cars)
self.assertEqual(garage.cars, {})
def test_init_add_car_garage_instance(self):
cesar = Cesar(rnd_name())
garage = Garage(rnd_town(), rnd_int(), rnd_uuid())
car = Car(rnd_float(), rnd_car_type(), rnd_car_producer(), rnd_float())
for value in [1000, 'some string', [garage]]:
with self.assertRaises(TypeError):
cesar.add_car(car, value)
def test_not_equal_register_id(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car3 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), None, [car1, car2])
garage2 = Garage(rnd_town(), rnd_int(), None, [car3])
name = rnd_name()
garages = [garage1, garage2]
register_id = rnd_uuid()
cesar1 = Cesar(name, garages, register_id)
cesar2 = Cesar(name, garages, rnd_uuid())
self.assertNotEqual(cesar1, cesar2)
def test_add_same_car(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1])
before_cars = garage.cars
self.assertFalse(garage.add(car1))
after_cars = garage.cars
self.assertEqual(before_cars, after_cars)
def test_remove_instance(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
garage = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1])
with self.assertRaises(TypeError):
garage.add('My Car')
with self.assertRaises(TypeError):
garage.add([car1])
def test_init_add_car_emptiest_garage(self):
garage1 = Garage(rnd_town(), 1, rnd_uuid())
garage2 = Garage(rnd_town(), 2, rnd_uuid())
garage3 = Garage(rnd_town(), 3, rnd_uuid())
cesar = Cesar(rnd_name(), [garage1, garage2, garage3])
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
car2 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
cesar.add_car(car1)
self.assertEqual(garage1.places_count(), 1)
self.assertEqual(garage2.places_count(), 2)
self.assertEqual(garage3.places_count(), 2)
cesar.add_car(car2)
self.assertEqual(garage1.places_count(), 1)
self.assertEqual(garage2.places_count(), 1)
self.assertEqual(garage3.places_count(), 2)
def test_comparison(self):
car1 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car2 = Car(2000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1, car2])
cesar1 = Cesar(rnd_name(), [garage1])
car1 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car2 = Car(3000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1, car2])
cesar2 = Cesar(rnd_name(), [garage1])
car1 = Car(2000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
car2 = Car(1000.0, rnd_car_type(), rnd_car_producer(), rnd_float())
garage1 = Garage(rnd_town(), rnd_int(), rnd_uuid(), [car1, car2])
cesar3 = Cesar(rnd_name(), [garage1])
self.assertGreater(cesar2, cesar1)
self.assertGreaterEqual(cesar1, cesar3)
self.assertLess(cesar1, cesar2)
self.assertLessEqual(cesar1, cesar3)
def test_init_with_invalid_price(self):
for value in [1000, '5000', '111,23']:
with self.assertRaises(ValueError):
Car(value, rnd_car_type(), rnd_float(), rnd_float())
def test_set_number(self):
car = Car(rnd_float(), rnd_car_type(), rnd_car_producer(), rnd_float())
new_number = rnd_uuid()
car.set_number(new_number)
self.assertEqual(car.number, new_number)
self.assertIsInstance(car.number, uuid.UUID)
def test_init_with_invalid_cars(self):
car1 = Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float())
for value in ['My Car', car1, {'car': car1}]:
with self.assertRaises(ValueError):
Garage(rnd_town(), rnd_int(), rnd_uuid(), value)
def test_init_with_invalid_mileage(self):
for value in [1000, '5000', '111,23']:
with self.assertRaises(ValueError):
Car(rnd_float(), rnd_car_type(), rnd_car_producer(), value)
def test_init_with_invalid_car_producer(self):
for value in ['My Producer', 1000, ['Producer 1', 'Producer 2']]:
with self.assertRaises(InvalidValueException):
Car(rnd_float(), rnd_car_type(), value, rnd_float())
def test_init_with_invalid_number(self):
for value in ['some string', 1000]:
with self.assertRaises(ValueError):
Car(rnd_float(), rnd_car_type(), rnd_car_producer(),
rnd_float(), value)
import random
from tests_homework.constants import CARS_TYPES, CARS_PRODUCER, TOWNS
from tests_homework.models import Cesar, Garage, Car
from tests_homework.utils import separator, debug_instance_json, \
debug_instance_pickle, debug_instance_yaml
if __name__ == '__main__':
car1 = Car(price=random.uniform(5000, 50000),
car_type=random.choice(CARS_TYPES),
producer=random.choice(CARS_PRODUCER),
mileage=random.uniform(1000, 5000))
car2 = Car(price=random.uniform(5000, 50000),
car_type=random.choice(CARS_TYPES),
producer=random.choice(CARS_PRODUCER),
mileage=random.uniform(1000, 5000))
car3 = Car(price=random.uniform(5000, 50000),
car_type=random.choice(CARS_TYPES),
producer=random.choice(CARS_PRODUCER),
mileage=random.uniform(1000, 5000))
car4 = Car(price=random.uniform(5000, 50000),
car_type=random.choice(CARS_TYPES),
producer=random.choice(CARS_PRODUCER),
mileage=random.uniform(1000, 5000))
garage1 = Garage(town=random.choice(TOWNS), places=2)
garage2 = Garage(town=random.choice(TOWNS), places=3)
garage3 = Garage(town=random.choice(TOWNS), places=4)
cesar1 = Cesar(name="John")
cesar1.add_car(car1, garage1)