def test_define_sample_points_nsample_lte_nsimu(self, mock_rand):
PI = PredictionIntervals()
aa = np.random.rand([400, 1])
mock_rand.return_value = aa
iisample, nsample = PI._define_sample_points(nsample=400, nsimu=500)
self.assertTrue(np.array_equal(iisample,
np.ceil(aa * 500) - 1),
msg='Expect range(500)')
self.assertEqual(nsample, 400, msg='Expect nsample to stay 400')
def test_define_sample_points_nsample_gt_nsimu(self):
PI = PredictionIntervals()
iisample, nsample = PI._define_sample_points(nsample=1000, nsimu=500)
self.assertEqual(iisample, range(500), msg='Expect range(500)')
self.assertEqual(nsample, 500, msg='Expect nsample updated to 500')
