def testSummary(self):
loss_array = []
for i in range(10):
loss_model = loss.Loss(FixedValueModel(1), FixedValueModel(i))
loss_array += loss_model.simulate_years(100)
loss_array += loss_model.simulate_years(
10) # Add a few more for value 9
summary = loss_model.summarize_loss(loss_array)
self.assertEqual(summary['minimum'], 0)
self.assertEqual(summary['tenth_percentile'], 1)
self.assertEqual(summary['mode'], 9)
self.assertEqual(summary['median'], 5)
self.assertEqual(summary['ninetieth_percentile'], 9)
self.assertEqual(summary['maximum'], 9)
def test_simulate_years_frequency(self):
loss_model = loss.Loss(FixedValueModel(2), FixedValueModel(0.5))
multiple_years = loss_model.simulate_years(3)
self.assertEqual([1.0, 1.0, 1.0], multiple_years)
def test_simulate_zeros(self):
loss_model = loss.Loss(FixedValueModel(0), FixedValueModel(0.5))
multiple_years = loss_model.simulate_years(3)
self.assertEqual([0, 0, 0], multiple_years)
def test_simulate_losses_one_year_multiple_loss(self):
loss_model = loss.Loss(FixedValueModel(3), FixedValueModel(0.5))
single_loss = loss_model.simulate_losses_one_year()
self.assertEqual([0.5, 0.5, 0.5], single_loss)
