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")