def test_marker_rotate_by_theta(self):
# rotate by 90
m = Marker(1, 0, 1, 0)
m.rotate_by_theta(2 * math.pi * 90 / 360, 0, 0)
self.assertEqual(m.get_index(), 1)
self.assertEqual(m.get_x(), -1)
self.assertTrue(math.fabs(m.get_y()) < 0.00001)
self.assertEqual(m.get_z(), 0)
# rotate by -90
m.rotate_by_theta(2 * math.pi * -90 / 360, 0, 0)
self.assertEqual(m.get_index(), 1)
self.assertEqual(m.get_x(), 0)
self.assertTrue(math.fabs(m.get_y() - 1) < 0.00001)
self.assertEqual(m.get_z(), 0)
# rotate by 45
m.rotate_by_theta(2 * math.pi * 45 / 360, 0, 0)
self.assertEqual(m.get_index(), 1)
self.assertTrue(math.fabs(m.get_x() + 0.7071) < 0.1)
self.assertTrue(math.fabs(m.get_y() - 0.7071) < 0.00001)
self.assertEqual(m.get_z(), 0)
def test_marker_3dmarker_format(self):
m = Marker(None, None, None, None)
self.assertEqual(m.get_3dmarker_format(),
'')
m = Marker(1, None, None, None)
self.assertEqual(m.get_3dmarker_format(),
'')
m = Marker(1, 1, None, None)
self.assertEqual(m.get_3dmarker_format(),
'')
m = Marker(1, 1, 1, None)
self.assertEqual(m.get_3dmarker_format(),
'')
m = Marker(1, 2, 3, 4)
self.assertEqual(m.get_3dmarker_format(),
' 1 2.000000 3.000000 4.000000')
m = Marker(10, -20, -30, -400)
self.assertEqual(m.get_3dmarker_format(),
' 10 -20.000000 -30.000000 -400.000000')
m = Marker(10, -20, -30, -4000)
self.assertEqual(m.get_3dmarker_format(),
' 10 -20.000000 -30.000000 -4000.000000')
m = Marker(1000, -20, -30, -40000)
self.assertEqual(m.get_3dmarker_format(),
' 1000 -20.000000 -30.000000 -40000.000000')
def test_marker_shift_marker(self):
m = Marker(1, 2, 3, 4)
m.shift_marker(0, 0, 0)
self.assertEqual(m.get_index(), 1)
self.assertEqual(m.get_x(), 2)
self.assertEqual(m.get_y(), 3)
self.assertEqual(m.get_z(), 4)
m.shift_marker(1, -1, 2)
self.assertEqual(m.get_index(), 1)
self.assertEqual(m.get_x(), 3)
self.assertEqual(m.get_y(), 2)
self.assertEqual(m.get_z(), 6)
def test_marker_constructor(self):
m = Marker(1, 2, 3, 4)
self.assertEqual(m.get_index(), 1)
self.assertEqual(m.get_x(), 2)
self.assertEqual(m.get_y(), 3)
self.assertEqual(m.get_z(), 4)
