From b4d14b6fd521da78b1683b78f9b125fa5a4367d1 Mon Sep 17 00:00:00 2001 From: prmiles Date: Thu, 6 Jun 2019 09:35:28 -0400 Subject: [PATCH] added test for dram output --- test/test_dram.py | 29 ++++++++++++++++++++++++++++- 1 file changed, 28 insertions(+), 1 deletion(-) diff --git a/test/test_dram.py b/test/test_dram.py index 17401d1..58ec932 100644 --- a/test/test_dram.py +++ b/test/test_dram.py @@ -445,11 +445,38 @@ class NLDRAM(unittest.TestCase): return_value=Calc) def test_io(self, mock_1): tmp = setup_problem() + iters = 2000 + burn = 1000 + thin = 1 + adapt = 200 + n_keep = np.floor_divide(iters - burn - 1, thin) + 1 + q = tmp['q'] + L = tmp['L'] + varS1 = tmp['varS1'] paramList, variables = tmp['paramList'], tmp['variables'] init_z, lower, upper = tmp['init_z'], tmp['lower'], tmp['upper'] a = dram.nlDRAM(None, paramList, variables, init_z, lower, upper, - plot=False, iters=2000, burn=1000, adapt=200) + plot=False, iters=iters, burn=burn, thin=thin, + adapt=adapt) self.assertTrue(isinstance(a, tuple), msg='Expect tuple return') self.assertEqual(len(a), 8, msg='Expect tuple of length 8') + self.assertEqual(a[0].shape, (n_keep, q), + msg='Expect (n_keep, q) array') + self.assertEqual(a[1], n_keep, + msg='Expect n_keep') + self.assertEqual(a[2].shape, varS1.shape, + msg='Expect matching shape') + self.assertEqual(a[3].shape, (iters-burn,), + msg='Expect (iters-burn,) array') + self.assertEqual(a[4].shape, (iters-burn, L), + msg='Expect (iters-burn, L) array') + self.assertTrue(isinstance(a[5], float), + msg=str('Expect float - got {}'.format(type(a[5])))) + self.assertTrue(isinstance(a[6], np.ndarray), + msg=str('Expect array - got {}'.format(type(a[6])))) + self.assertTrue(isinstance(a[7], np.ndarray), + msg=str('Expect array - got {}'.format(type(a[7])))) + + \ No newline at end of file