def test9(self): print "Testing BSE states" instance = BSE() stars = Particles(2) stars.mass = [1.0, 0.2] | units.MSun binaries = Particles(1) orbital_period = 200.0 | units.day semi_major_axis = instance.orbital_period_to_semi_major_axis( orbital_period, stars[0].mass, stars[1].mass) binaries.semi_major_axis = semi_major_axis binaries.eccentricity = 0.0 binaries[0].child1 = stars[0] binaries[0].child2 = stars[1] print "First do everything manually:", self.assertEquals(instance.get_name_of_current_state(), 'UNINITIALIZED') instance.initialize_code() self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED') instance.commit_parameters() self.assertEquals(instance.get_name_of_current_state(), 'RUN') instance.cleanup_code() self.assertEquals(instance.get_name_of_current_state(), 'END') instance.stop() print "ok" print "initialize_code(), commit_parameters(), " \ "and cleanup_code() should be called automatically:", instance = BSE() self.assertEquals(instance.get_name_of_current_state(), 'UNINITIALIZED') instance.parameters.reimers_mass_loss_coefficient = 0.5 self.assertEquals(instance.get_name_of_current_state(), 'INITIALIZED') instance.particles.add_particles(stars) instance.binaries.add_particles(binaries) self.assertEquals(instance.get_name_of_current_state(), 'RUN') instance.stop() self.assertEquals(instance.get_name_of_current_state(), 'STOPPED') print "ok"
def test9(self): print "Testing BSE states" instance = BSE() stars = Particles(2) stars.mass = [1.0, 0.2] | units.MSun binaries = Particles(1) orbital_period = 200.0 | units.day semi_major_axis = instance.orbital_period_to_semi_major_axis(orbital_period, stars[0].mass, stars[1].mass) binaries.semi_major_axis = semi_major_axis binaries.eccentricity = 0.0 binaries[0].child1 = stars[0] binaries[0].child2 = stars[1] print "First do everything manually:", self.assertEquals(instance.get_name_of_current_state(), "UNINITIALIZED") instance.initialize_code() self.assertEquals(instance.get_name_of_current_state(), "INITIALIZED") instance.commit_parameters() self.assertEquals(instance.get_name_of_current_state(), "RUN") instance.cleanup_code() self.assertEquals(instance.get_name_of_current_state(), "END") instance.stop() print "ok" print "initialize_code(), commit_parameters(), " "and cleanup_code() should be called automatically:", instance = BSE() self.assertEquals(instance.get_name_of_current_state(), "UNINITIALIZED") instance.parameters.reimers_mass_loss_coefficient = 0.5 self.assertEquals(instance.get_name_of_current_state(), "INITIALIZED") instance.particles.add_particles(stars) instance.binaries.add_particles(binaries) self.assertEquals(instance.get_name_of_current_state(), "RUN") instance.stop() self.assertEquals(instance.get_name_of_current_state(), "STOPPED") print "ok"