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