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_write_markers_to_file(self):
temp_dir = tempfile.mkdtemp()
try:
# invalid filename
markers = MarkersList()
try:
markers.write_markers_to_file(temp_dir)
self.fail('Expected IOError')
except IOError:
pass
emptyfile = os.path.join(temp_dir, 'emptyfile')
# empty markers object
markers.write_markers_to_file(emptyfile)
self.assertTrue(os.path.isfile(emptyfile))
self.assertEqual(os.path.getsize(emptyfile), 0)
# write 1 marker
markers.add_marker(1, 2, 3, 4)
one = os.path.join(temp_dir, 'one')
markers.write_markers_to_file(one)
self.assertTrue(os.path.isfile(one))
f = open(one, 'r')
line = f.readline()
self.assertEqual(line,
' 1 2.000000 3.000000 4.000000\n')
f.close()
# write 2 markers
markers.add_marker(2, 3, 4, 5)
two = os.path.join(temp_dir, 'two')
markers.write_markers_to_file(two)
self.assertTrue(os.path.isfile(two))
f = open(two, 'r')
line = f.readline()
self.assertEqual(line,
' 1 2.000000 3.000000 4.000000\n')
line = f.readline()
self.assertEqual(line,
' 2 3.000000 4.000000 5.000000\n')
f.close()
finally:
shutil.rmtree(temp_dir)
