def test_carDrive_whenNotEnoughFuel_shouldRaise(self):
make = 'test make'
model = 'test model'
fuel_consumption = 6
fuel_capacity = 60
params = [make, model, fuel_consumption, fuel_capacity]
car = Car(*params)
with self.assertRaises(Exception) as context:
car.drive(100)
self.assertIsNotNone(context.exception)
def test_carDrive_whenEnoughFuel_shouldDecreaseFuel(self):
make = 'test make'
model = 'test model'
fuel_consumption = 6
fuel_capacity = 60
params = [make, model, fuel_consumption, fuel_capacity]
car = Car(*params)
car.refuel(car.fuel_capacity)
distance = 100
car.drive(distance)
expected = car.fuel_capacity - car.fuel_consumption * distance / 100
actual = car.fuel_amount
self.assertEqual(expected, actual)