def test_check_pi_flag_false_none(self): PI = PredictionIntervals() prediction_intervals = PI._check_prediction_interval_flag( plot_pred_int=False, prediction_intervals=None) self.assertEqual(prediction_intervals, None, msg='Expect None')
def test_check_pi_flag_true_0(self): PI = PredictionIntervals() prediction_intervals = PI._check_prediction_interval_flag( plot_pred_int=True, prediction_intervals=0) self.assertEqual(prediction_intervals, 0, msg='Expect 0')