def tests_refuel_method__when_fuel_equal_to_capacity__should_work(self):
test_vehicle = Vehicle(self.fuel, self.horse_power)
test_vehicle.drive(8)
test_vehicle.refuel(10)
result = test_vehicle.fuel
expected = 10
self.assertEqual(expected, result)
def tests_refuel_method__when_fuel_less_than_capacity__should_work(self):
test_vehicle = Vehicle(self.fuel, self.horse_power)
test_vehicle.drive(5)
test_vehicle.refuel(2)
result = test_vehicle.fuel
expected = 5.75
self.assertEqual(expected, result)
def tests_refuel_method__when_fuel_more_than_capacity__should_raise_exception(
self):
test_vehicle = Vehicle(self.fuel, self.horse_power)
with self.assertRaises(Exception) as result:
test_vehicle.refuel(5)
self.assertEqual(result.exception.args[0], "Too much fuel")