def test_data_selection(self): incl_temps, n_add = lib_add.get_n_add(self.temps, self.starting_temps, self.tmax) for num, temp in enumerate(incl_temps): #print temp, self.ref_incl_temps[num] self.assertEqual(temp, self.ref_incl_temps[num]) self.assertEqual(n_add, self.ref_n_add)
def get_n_add(self): n_add = lib_add.get_n_add(self.add_checks_temperatures, self.add_checks_starting_temperatures, self.tmax)[1] self.pars['n_add'] = n_add return n_add