added test for dram output

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]))))
+
+