def test_generate_chain_list_with_non_default_burnin(self):
pres = setup_pres()
chains = CP.generate_chain_list(pres=pres, burnin_percentage=0)
self.assertEqual(len(chains), 2, msg='expect length of 2')
self.assertTrue(np.array_equal(chains[0], pres[0]['chain']))
self.assertTrue(np.array_equal(chains[1], pres[1]['chain']))
def test_generate_chain_list(self):
pres = setup_pres()
chains = CP.generate_chain_list(pres=pres)
self.assertEqual(len(chains), 2, msg='expect length of 2')
self.assertTrue(np.array_equal(chains[0], pres[0]['chain'][50:, :]))
self.assertTrue(np.array_equal(chains[1], pres[1]['chain'][50:, :]))