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')