Ejemplo n.º 1
0
    def test3(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)
        index2, error = instance.new_particle(mass=0.001,
                                              radius=0,
                                              x=3.0,
                                              y=4.0,
                                              z=5.0,
                                              vx=3.3,
                                              vy=4.4,
                                              vz=5.5)
        self.assertEqual(error, 0)

        self.assertEqual(1, index1)
        self.assertEqual(2, index2)

        mass, error = instance.get_mass(1)
        self.assertEqual(error, 0)
        self.assertEqual(mass, 1.0)

        mass, error = instance.get_mass(2)
        self.assertEqual(error, 0)
        self.assertEqual(mass, 0.001)

        error = instance.delete_particle(1)
        mass, error = instance.get_mass(1)
        self.assertEqual(error, -1)

        index1, error = instance.new_particle(mass=2.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(1, index1)

        mass, error = instance.get_mass(1)
        self.assertEqual(error, 0)
        self.assertEqual(mass, 2.0)

        mass, error = instance.get_mass(2)
        self.assertEqual(error, 0)
        self.assertEqual(mass, 0.001)
Ejemplo n.º 2
0
    def test3(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)
        index2, error = instance.new_particle(mass = 0.001, radius = 0, x = 3.0, y = 4.0, z = 5.0, vx = 3.3, vy = 4.4, vz = 5.5)
        self.assertEquals(error, 0)
        
        self.assertEquals(1, index1)
        self.assertEquals(2, index2)
        
        mass,error = instance.get_mass(1)
        self.assertEquals(error, 0)
        self.assertEquals(mass, 1.0)
                
        mass,error = instance.get_mass(2)
        self.assertEquals(error, 0)
        self.assertEquals(mass, 0.001)
        
        error  = instance.delete_particle(1)
        mass,error = instance.get_mass(1)
        self.assertEquals(error, -1)
        
        index1, error = instance.new_particle(mass = 2.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(1, index1)
        
        mass,error = instance.get_mass(1)
        self.assertEquals(error, 0)
        self.assertEquals(mass, 2.0)
                
        mass,error = instance.get_mass(2)
        self.assertEquals(error, 0)
        self.assertEquals(mass, 0.001)