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_drive_method__when_fuel_less_than_fuel_needed_should_raise_exception(
self):
test_vehicle = Vehicle(self.fuel, self.horse_power)
with self.assertRaises(Exception) as result:
test_vehicle.drive(15)
self.assertEqual(result.exception.args[0], "Not enough fuel")
def tests_drive_method__when_fuel_equal_to_fuel_needed__should_work(self):
test_vehicle = Vehicle(self.fuel, self.horse_power)
test_vehicle.drive(8)
result = test_vehicle.fuel
expected = 0
self.assertEqual(expected, result)