def test_csv_single_false(self): csv = db.csv("UnitTest_tmp", self.parnames, self.like, self.randompar, simulations=self.simulations, chains=1, save_sim=False) csv.save(self.like, self.randompar, self.simulations) csv.finalize() csvdata = csv.getdata() self.assertEqual(str(type(csvdata)), "<class 'numpy.ndarray'>") self.assertEqual(len(csvdata[0]), 7) self.assertEqual(len(csvdata), 2) self.assertEqual(len(csv.header), 7)
def test_csv_multiline(self): csv = db.csv("UnitTest_tmp", self.parnames, self.like, self.randompar, simulations=self.simulations_multi, chains=1, save_sim=True) csv.save(self.like ,self.randompar,self.simulations_multi) # Save Simulations csv.finalize() csvdata = csv.getdata() self.assertEqual(str(type(csvdata)), "<class 'numpy.ndarray'>") self.assertEqual(len(csvdata[0]), 32) self.assertEqual(len(csvdata), 2) self.assertEqual(len(csv.header), 32)
def test_csv_single(self): csv = db.csv("UnitTest_tmp", self.parnames, self.like, self.randompar, simulations=self.simulations, chains=1, save_sim=True) csv.save(self.like, self.randompar, self.simulations) csv.finalize() csvdata = csv.getdata() self.assertEqual(str(type(csvdata)), str(type(np.array([])))) self.assertEqual(len(csvdata[0]), 12) self.assertEqual(len(csvdata), 2) self.assertEqual(len(csv.header), 12)
def test_csv_multiline_false(self): # Save not Simulations csv = db.csv("UnitTest_tmp", self.parnames, self.like, self.randompar, simulations=self.simulations_multi, chains=1, save_sim=False) csv.save(self.like, self.randompar, self.simulations_multi) csv.finalize() csvdata = csv.getdata() self.assertEqual(str(type(csvdata)), str(type(np.array([])))) self.assertEqual(len(csvdata[0]), 7) self.assertEqual(len(csvdata), 2) self.assertEqual(len(csv.header), 7)