Python PredictionIntervals._analyze_data_structure Examples

Example #1

File: test_PredictionIntervals.py Project: williamqzy/pymcmcstat

 def test_basic_ds_shape_0(self):
     PI = PredictionIntervals()
     DS = gf.basic_data_structure()
     DS.shape = [(100, )]  # remove column - code should add it back
     ndatabatches, ncols = PI._analyze_data_structure(data=DS)
     self.assertEqual(ndatabatches, 1, msg='Expect 1 batch')
     self.assertEqual(ncols, [1], msg='Expect [1]')

Example #2

File: test_PredictionIntervals.py Project: williamqzy/pymcmcstat

 def test_ds_2_nbatch(self):
     PI = PredictionIntervals()
     DS = gf.non_basic_data_structure()
     ndatabatches, ncols = PI._analyze_data_structure(data=DS)
     self.assertEqual(ndatabatches, 2, msg='Expect 2 batches')
     self.assertEqual(ncols, [1, 2], msg='Expect [1, 1]')

Example #3

File: test_PredictionIntervals.py Project: williamqzy/pymcmcstat

 def test_basic_ds(self):
     PI = PredictionIntervals()
     DS = gf.basic_data_structure()
     ndatabatches, ncols = PI._analyze_data_structure(data=DS)
     self.assertEqual(ndatabatches, 1, msg='Expect 1 batch')
     self.assertEqual(ncols, [1], msg='Expect [1]')