def testUndo(self): apartamentController = ApartamentController() apartamentController.addBill(0,BillType.Water,100) apartamentController.undo() self.assertEqual(apartamentController.getBillsCostOfType(BillType.Water),0) apartamentController.addBill(0,BillType.Water,50) apartamentController.addBill(0,BillType.Water,100) apartamentController.addBill(0,BillType.Water,200) apartamentController.addBill(0,BillType.Water,300) apartamentController.undo() apartamentController.undo() self.assertEqual(apartamentController.getBillsCostOfType(BillType.Water),150)
def testGetBillsCostOfType(self): apartamentController = ApartamentController() apartamentController.addBill(0,BillType.Water,100) apartamentController.addBill(0,BillType.Gas,10) apartamentController.addBill(1,BillType.Water,100) apartamentController.addBill(1,BillType.Gas,10) apartamentController.addBill(2,BillType.Water,100) apartamentController.addBill(2,BillType.Water,5) apartamentController.addBill(3,BillType.Gas,80) apartamentController.addBill(4,BillType.Water,100) apartamentController.addBill(4,BillType.Water,5) self.assertEqual(apartamentController.getBillsCostOfType(BillType.Gas),100)