def test_return_max_amount(self):
contract = Contract(self.contract_data)
event = Event(event_id=1, deal_id=1, loss=3001)
result = contract.event_loss_reimbursement(event=event)
self.assertEqual(3000, result)
def test_return_event_loss_when_less_than_max_amount(self):
contract = Contract(self.contract_data)
event = Event(event_id=1, deal_id=1, loss=2999)
result = contract.event_loss_reimbursement(event=event)
self.assertEqual(2999, result)