Ejemplo n.º 1
0
    def test4(self):
        instance = MikkolaInterface()
        instance.initialize_code()
        instance.set_lightspeed(1e4)
        instance.commit_parameters()

        index1, error = instance.new_particle(mass=1.0,
                                              radius=0,
                                              x=1.0,
                                              y=2.0,
                                              z=3.0,
                                              vx=1.1,
                                              vy=2.2,
                                              vz=3.3)
        self.assertEqual(error, 0)
        self.assertEqual(index1, 1)

        instance.commit_particles()

        error = instance.cleanup_code()
        self.assertEqual(error, 0)

        mass, error = instance.get_mass(1)
        self.assertEqual(error, -1)
        instance.commit_parameters()
        index1, error = instance.new_particle(mass=1.0,
                                              radius=0,
                                              x=1.0,
                                              y=2.0,
                                              z=3.0,
                                              vx=1.1,
                                              vy=2.2,
                                              vz=3.3)
        self.assertEqual(error, 0)
        self.assertEqual(index1, 1)
        instance.commit_particles()
        error = instance.cleanup_code()
        self.assertEqual(error, 0)
Ejemplo n.º 2
0
    def test4(self):
        instance = MikkolaInterface()
        instance.initialize_code()
        instance.set_lightspeed(1e4)
        instance.commit_parameters()

        index1, error = instance.new_particle(mass = 1.0, radius = 0, x = 1.0, y = 2.0, z = 3.0, vx = 1.1, vy = 2.2, vz = 3.3)
        self.assertEquals(error, 0)
        self.assertEquals(index1, 1)
        
        instance.commit_particles()
        
        error = instance.cleanup_code()
        self.assertEquals(error, 0)
        
        mass,error = instance.get_mass(1)
        self.assertEquals(error, -1)
        instance.commit_parameters()
        index1, error = instance.new_particle(mass = 1.0, radius = 0, x = 1.0, y = 2.0, z = 3.0, vx = 1.1, vy = 2.2, vz = 3.3)
        self.assertEquals(error, 0)
        self.assertEquals(index1, 1)
        instance.commit_particles()
        error = instance.cleanup_code()
        self.assertEquals(error, 0)