def test_add_batch_column_title_ny_gt_1(self):
PI = PredictionIntervals()
htmp, ax = PI._initialize_plot_features(ii=0,
jj=0,
ny=1,
figsizeinches=[10, 11])
PI._add_batch_column_title(nbatch=1, ny=2, ii=0, jj=0)
self.assertEqual(ax.get_title(),
'Column #0',
msg='Strings should match')
self.assertEqual(htmp.get_figwidth(), 10.0, msg='Expect 10.0')
self.assertEqual(htmp.get_figheight(), 11.0, msg='Expect 11.0')
def test_initialize_plot_features(self):
PI = PredictionIntervals()
htmp, ax = PI._initialize_plot_features(ii=0,
jj=0,
ny=1,
figsizeinches=[10, 11])
self.assertEqual(htmp.get_figwidth(), 10.0, msg='Figure width is 10in')
self.assertEqual(htmp.get_figheight(),
11.0,
msg='Figure height is 11in')
self.assertEqual(htmp.get_label(),
'Batch # 0 | Column # 0',
msg='Strings should match')
self.assertEqual(ax.get_label(), '', msg='Expect empty string')