コード例 #1
0
 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)
コード例 #2
0
 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)