def test_markers_rotate_by_angle_zero_offset(self): markers = MarkersList() markers.add_marker(1, 1, 0, 2) markers.rotate_by_angle(90, 0, 0) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertTrue(m.get_x() < 0.00001) self.assertEqual(m.get_y(), 1) self.assertEqual(m.get_z(), 2) # this is does nothing markers.rotate_by_angle(0, 0, 0) markers.add_marker(2, 1, 0, 3) markers.rotate_by_angle(90, 0, 0) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertTrue(m.get_x() < 0.00001) self.assertTrue(m.get_y() < 0.00001) self.assertEqual(m.get_z(), 2) m = markers.get_markers()[1] self.assertEqual(m.get_index(), 2) self.assertTrue(m.get_x() < 0.00001) self.assertEqual(m.get_y(), 1) self.assertEqual(m.get_z(), 3)
def test_rotate_markers_file_outfile_set_to_none(self): temp_dir = tempfile.mkdtemp() try: markerfile = os.path.join(temp_dir, '3Dmarkers.txt') markers = MarkersList() markers.add_marker(1, 2, 3, 4) markers.write_markers_to_file(markerfile) theargs = Parameters() theargs.outfile = None theargs.angle = 90 theargs.width = 10 theargs.height = 10 theargs.markerfile = markerfile rotate_3dmarkers.rotate_markers_file(theargs) origfile = markerfile + '.orig' self.assertTrue(os.path.isfile(origfile)) fac = MarkersFrom3DMarkersFileFactory(origfile) markers = fac.get_markerslist() m = markers.get_markers()[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) self.assertTrue(os.path.isfile(markerfile)) fac = MarkersFrom3DMarkersFileFactory(markerfile) markers = fac.get_markerslist() m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertEqual(m.get_x(), 7) self.assertEqual(m.get_y(), 2) self.assertEqual(m.get_z(), 4) finally: shutil.rmtree(temp_dir)
def test_markers_rotate_by_angle_with_offset(self): markers = MarkersList() markers.add_marker(1, 1, 1, 5) markers.rotate_by_angle(90, 5, 5) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertTrue(math.fabs(m.get_x() - 9) < 0.0001) self.assertTrue(math.fabs(m.get_y() - 1) < 0.0001) self.assertEqual(m.get_z(), 5)
def test_markers_rotate_by_angle_none_offset(self): markers = MarkersList() markers.add_marker(1, 1, 0, 2) markers.rotate_by_angle(90, None, None) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertTrue(m.get_x() < 0.00001) self.assertEqual(m.get_y(), 1) self.assertEqual(m.get_z(), 2)
def test_markers_shift(self): markers = MarkersList() markers.shift_markers(10, 20, 30) self.assertEqual(len(markers.get_markers()), 0) markers.add_marker(1, 2, 3, 4) markers.shift_markers(10, 20, 30) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertEqual(m.get_x(), 12) self.assertEqual(m.get_y(), 23) self.assertEqual(m.get_z(), 34) markers.add_marker(2, 3, 4, 5) markers.shift_markers(10, 20, 30) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertEqual(m.get_x(), 22) self.assertEqual(m.get_y(), 43) self.assertEqual(m.get_z(), 64) m = markers.get_markers()[1] self.assertEqual(m.get_index(), 2) self.assertEqual(m.get_x(), 13) self.assertEqual(m.get_y(), 24) self.assertEqual(m.get_z(), 35) markers.shift_markers(0, 0, 0) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertEqual(m.get_x(), 22) self.assertEqual(m.get_y(), 43) self.assertEqual(m.get_z(), 64) m = markers.get_markers()[1] self.assertEqual(m.get_index(), 2) self.assertEqual(m.get_x(), 13) self.assertEqual(m.get_y(), 24) self.assertEqual(m.get_z(), 35) markers.shift_markers(-10, -20, -30) m = markers.get_markers()[0] self.assertEqual(m.get_index(), 1) self.assertEqual(m.get_x(), 12) self.assertEqual(m.get_y(), 23) self.assertEqual(m.get_z(), 34) m = markers.get_markers()[1] self.assertEqual(m.get_index(), 2) self.assertEqual(m.get_x(), 3) self.assertEqual(m.get_y(), 4) self.assertEqual(m.get_z(), 5)
def test_markers_add_get_clear(self): markers = MarkersList() self.assertEqual(len(markers.get_markers()), 0) markers.add_marker(1, 2, 3, 4) self.assertEqual(len(markers.get_markers()), 1) m = markers.get_markers()[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) markers.add_marker(2, 3, 4, 5) self.assertEqual(len(markers.get_markers()), 2) m = markers.get_markers()[1] self.assertEqual(m.get_index(), 2) self.assertEqual(m.get_x(), 3) self.assertEqual(m.get_y(), 4) self.assertEqual(m.get_z(), 5) markers.clear_markers() self.assertEqual(len(markers.get_markers()), 0)