def test16(self): instance = Huayno() instance.parameters.epsilon_squared = 0.0 | nbody_system.length**2 particles = datamodel.Particles(2) particles.mass = [1.] | nbody_system.mass particles.radius = [0.0] | nbody_system.length particles.position = [[0.0,0.0,0.0],[1.0,0.0,0.0]] | nbody_system.length particles.velocity = [[0.0, 0.0, 0.0]] | nbody_system.speed instance.particles.add_particles(particles) zero = 0.0 | nbody_system.length for x in (0.25, 0.5, 0.75): x0 = x | nbody_system.length potential0 = instance.get_potential_at_point(zero, x0, zero, zero) fx0, fy0, fz0 = instance.get_gravity_at_point(zero, x0, zero, zero) self.assertAlmostEqual(fy0, 0.0 | nbody_system.acceleration,14) self.assertAlmostEqual(fz0, 0.0 | nbody_system.acceleration,14) fx = (-1.0 / (x0**2)+1.0 / (((1.0|nbody_system.length)-x0)**2)) * (1.0 | nbody_system.length ** 3 / nbody_system.time ** 2) self.assertAlmostEqual(fx, fx0,14) self.assertAlmostEqual(potential0, -nbody_system.G*(1.|nbody_system.mass)*(1./x0+1./((1.|nbody_system.length)-x0)),14) instance.stop()
def test16(self): solsys = new_solar_system() solsys.x-=1.| units.AU p=datamodel.Particles(3) p.mass=[1,2,3] | units.MSun p.x=[1,10,100] | units.AU p.y=[0,0,-10] | units.AU p.z=[0,0,10] | units.AU from amuse.community.huayno.interface import Huayno from amuse.units import nbody_system conv=nbody_system.nbody_to_si(1. | units.AU, 1.| units.MSun) h = Huayno(conv) h.particles.add_particles(solsys) ax1,ay1,az1=h.get_gravity_at_point(p.x*0.,p.x,p.y,p.z) mercury = Mercury() mercury.particles.add_particles(solsys) ax2,ay2,az2=mercury.get_gravity_at_point(p.x*0.,p.x,p.y,p.z) self.assertAlmostRelativeEqual(ax1,ax2,12) self.assertAlmostRelativeEqual(ay1,ay2,12) self.assertAlmostRelativeEqual(az1,az2,12)
def test16(self): instance = Huayno() instance.parameters.epsilon_squared = 0.0 | nbody_system.length**2 particles = datamodel.Particles(2) particles.mass = [1.] | nbody_system.mass particles.radius = [0.0] | nbody_system.length particles.position = [[0.0, 0.0, 0.0], [1.0, 0.0, 0.0] ] | nbody_system.length particles.velocity = [[0.0, 0.0, 0.0]] | nbody_system.speed instance.particles.add_particles(particles) zero = 0.0 | nbody_system.length for x in (0.25, 0.5, 0.75): x0 = x | nbody_system.length potential0 = instance.get_potential_at_point(zero, x0, zero, zero) fx0, fy0, fz0 = instance.get_gravity_at_point(zero, x0, zero, zero) self.assertAlmostEqual(fy0, 0.0 | nbody_system.acceleration, 14) self.assertAlmostEqual(fz0, 0.0 | nbody_system.acceleration, 14) fx = (-1.0 / (x0**2) + 1.0 / (((1.0 | nbody_system.length) - x0)**2)) * ( 1.0 | nbody_system.length**3 / nbody_system.time**2) self.assertAlmostEqual(fx, fx0, 14) self.assertAlmostEqual( potential0, -nbody_system.G * (1. | nbody_system.mass) * (1. / x0 + 1. / ((1. | nbody_system.length) - x0)), 14) instance.stop()