def test3(self): print "Testing get/set for metallicity..." instance = EVtwinInterface() error = instance.initialize_code() self.assertEquals(0, error) error = instance.set_ev_path(instance.get_data_directory()) self.assertEquals(0, error) (metallicity, error) = instance.get_metallicity() self.assertEquals(0, error) self.assertEquals(0.02, metallicity) # Z > 0.04 is not allowed error = instance.set_metallicity(0.0401) self.assertEquals(0, error) error = instance.commit_parameters() self.assertEquals(-2, error) # 0.0 <= Z <= 0.04 is allowed any_allowed_Z = numpy.random.random() * 0.04 error = instance.set_metallicity(any_allowed_Z) self.assertEquals(0, error) (metallicity, error) = instance.get_metallicity() self.assertEquals(0, error) self.assertEquals(any_allowed_Z, metallicity) error = instance.commit_parameters() self.assertEquals(0, error) error = instance.cleanup_code() self.assertEquals(0, error) instance.stop()
def test1(self): instance = EVtwinInterface() error = instance.initialize_code() self.assertEquals(0, error) error = instance.cleanup_code() self.assertEquals(0, error) instance.stop()
def test2(self): instance = EVtwinInterface() error = instance.initialize_code() self.assertEquals(0, error) error = instance.set_ev_path(instance.get_data_directory()) self.assertEquals(0, error) error = instance.cleanup_code() self.assertEquals(0, error) instance.stop()
def test5(self): print "Testing basic operations (new_particle_method, evolve_one_step etc.)..." instance = EVtwinInterface() error = instance.initialize_code() self.assertEquals(0, error) error = instance.set_ev_path(instance.get_data_directory()) self.assertEquals(0, error) error = instance.commit_parameters() self.assertEquals(0, error) (index_of_the_star, error) = instance.new_particle_method(1.05) self.assertEquals(0, error) self.assertTrue(index_of_the_star >= 0) error = instance.commit_particles() self.assertEquals(0, error) error = instance.set_wind_multiplier(index_of_the_star, 1.0) self.assertEquals(0, error) (mass, error) = instance.get_mass(index_of_the_star) self.assertEquals(0, error) self.assertEquals(1.05, mass) for i in range(3): error = instance.evolve_one_step(index_of_the_star) self.assertEquals(0, error) (mass, error) = instance.get_mass(index_of_the_star) self.assertEquals(0, error) self.assertTrue(mass <= 1.05) (age, error) = instance.get_age(index_of_the_star) self.assertEquals(0, error) self.assertTrue(age > 0) (x, error) = instance.get_wind_mass_loss_rate(index_of_the_star) self.assertEquals(0, error) self.assertTrue(x < 1e-13, "mass loss should be less than 1e-13, it is {0}".format(x)) self.assertTrue(x > 0, "mass loss rate should be more than 0, it is {0}".format(x)) error = instance.cleanup_code() self.assertEquals(0, error) instance.stop()
def test4(self): print "Testing get/set for maximum number of stars..." instance = EVtwinInterface() error = instance.initialize_code() self.assertEquals(0, error) error = instance.set_ev_path(instance.get_data_directory()) self.assertEquals(0, error) (value, error) = instance.get_maximum_number_of_stars() self.assertEquals(0, error) self.assertEquals(10, value) error = instance.set_maximum_number_of_stars(10000) self.assertEquals(0, error) (value, error) = instance.get_maximum_number_of_stars() self.assertEquals(0, error) self.assertEquals(10000, value) error = instance.commit_parameters() self.assertEquals(0, error) error = instance.cleanup_code() self.assertEquals(0, error) instance.stop()