def test_perpendicular(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield, pi/2)
xcomp = -self.vd / (1 + self.ot**2) * self.ot
ycomp = 0.0
zcomp = self.vd / (1 + self.ot**2)
self.assertAlmostEqual(vdv[0], xcomp)
self.assertAlmostEqual(vdv[1], ycomp)
self.assertAlmostEqual(vdv[2], zcomp)
def test_perpendicular(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield,
pi / 2)
xcomp = -self.vd / (1 + self.ot**2) * self.ot
ycomp = 0.0
zcomp = self.vd / (1 + self.ot**2)
self.assertAlmostEqual(vdv[0], xcomp)
self.assertAlmostEqual(vdv[1], ycomp)
self.assertAlmostEqual(vdv[2], zcomp)
def test_shape(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield,
self.tilt)
self.assertEqual(vdv.shape, (3, ))
def test_unit_vector_magnitude(self):
vdv = sim.drift_velocity_vector(1.0, self.efield, self.bfield,
self.tilt)
self.assertAlmostEqual(numpy.linalg.norm(vdv), 1.0, delta=1e-1)
def test_magnitude(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield,
self.tilt)
self.assertAlmostEqual(numpy.linalg.norm(vdv), self.vd, delta=0.1)
def test_no_bfield(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, 0, self.tilt)
self.assertEqual(vdv[0], 0.)
self.assertEqual(vdv[1], 0.)
def test_zero_angle(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield, 0)
self.assertEqual(vdv[0], 0.)
self.assertEqual(vdv[1], 0.)
def test_shape(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield, self.tilt)
self.assertEqual(vdv.shape, (3,))
def test_unit_vector_magnitude(self):
vdv = sim.drift_velocity_vector(1.0, self.efield, self.bfield, self.tilt)
self.assertAlmostEqual(numpy.linalg.norm(vdv), 1.0, delta=1e-1)
def test_magnitude(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield, self.tilt)
self.assertAlmostEqual(numpy.linalg.norm(vdv), self.vd, delta=0.1)
def test_no_bfield(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, 0, self.tilt)
self.assertEqual(vdv[0], 0.)
self.assertEqual(vdv[1], 0.)
def test_zero_angle(self):
vdv = sim.drift_velocity_vector(self.vd, self.efield, self.bfield, 0)
self.assertEqual(vdv[0], 0.)
self.assertEqual(vdv[1], 0.)
