def test_driveWhenNotEnoughtFuelRaiseException(self):
make = "Test Car"
model = 'Test model'
fuel_consumption = 6
fuel_capacity = 60
params = [make, model, fuel_consumption, fuel_capacity]
car = Car(*params)
with self.assertRaises(Exception) as x:
car.drive(100)
self.assertIsNotNone(x.exception)
def test_driveWhithEnoughtFuelRaiseException(self):
make = "Test Car"
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)