class TestEx7(unittest.TestCase): def setUp(self): self.gasoline_pump = GasolinePump(400, 3.00) def test_get_pump_volume(self): self.gasoline_pump.volume |should| equal_to(400) def test_fill_pump(self): self.gasoline_pump.fill() |should| equal_to(400) def test_supply_60_by_value_should_return_20(self): self.gasoline_pump.supply_by_value(60) |should| equal_to(20) def test_supply_by_value_in_pump_with_insufficient_volume(self): self.gasoline_pump.volume = 20 self.gasoline_pump.supply_by_value(90) |should| \ equal_to('volume unavailable') self.gasoline_pump.volume |should| equal_to(20) def test_supply_by_number_of_liters_20_should_return_60(self): self.gasoline_pump.supply_by_number_of_liters(20) |should| equal_to(60) def test_supply_by_number_of_liters_with_insufficient_volume(self): self.gasoline_pump.volume = 20 self.gasoline_pump.supply_by_number_of_liters(30) |should| \ equal_to('volume unavailable') self.gasoline_pump.volume |should| equal_to(20)
def setUp(self): self.gasoline_pump = GasolinePump(400, 3.00)