def test_write_map(self):
before = self.ds_map()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'type map',
'meta string:nat32 location 12 Middle Earth', 'element city 9',
'property string:nat32 name', 'property real32 x',
'property real32 y', 'element tower 3',
'property string:nat32 name', 'property real32 x',
'property real32 y', 'end_header', '8 Hobbiton 67.400002 55.5',
'4 Bree 79 54.099998', '9 Rivendell 99.099998 53.299999',
'5 Moria 100.5 69.5', u'11 Lothl\u00f3rien 113 74.599998',
'6 Edoras 105.1 99.599998', "11 Helm's Deep 99.5 98.699997",
'8 Isengard 98.5 94.800003', '12 Minas Tirith 135.60001 111.3',
'7 Orthanc 98.5 94.800003', u'10 Barad-d\u00fbr 156.2 107.8',
'12 Cirith Ungol 145.2 111', ''
]
self.assertTrue(data.decode('utf8') == '\n'.join(expected))
temp.close()
def test_write_read_graph(self):
before = self.ds_graph()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
after = ply2.read(temp)
temp.close()
self.equal(before, after)
def test_write_read_graph(self):
before = self.ds_graph()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
after = ply2.read(temp)
temp.close()
self.equal(before, after)
def test_write_meta(self):
before = self.ds_meta()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'meta string:nat32 author 7 Cthulhu', 'meta int64 tentacles 42', 'meta real64 pi 3.141592653589793', 'end_header', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_write_arrays(self):
before = self.ds_arrays()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'element samples 5', 'property array:1:nat32:int8 values', 'end_header', '4 3 1 4 2', '2 42 42', '6 100 101 102 -1 -2 0', '1 -12', '0 ', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_write_strings(self):
before = self.ds_strings()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'element people 8', 'property string:nat32 name', 'end_header', '9 The Alien', '6 Ripley', '3 Ash', '7 Bite Me', '4 ', '9 Penguin', '7 Joker ', '8 Two\nFace', '']
self.assertTrue(data=='\n'.join(expected).encode('utf8'))
temp.close()
def test_write_image(self):
before = self.ds_image()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'type image.rgb', 'element pixel 2 2', 'property nat8 red', 'property nat8 green', 'property nat8 blue', 'end_header', '0 0 0', '255 0 0', '0 255 0', '0 0 255', '']
self.assertTrue(data=='\n'.join(expected).encode('utf8'))
temp.close()
def test_write_map(self):
before = self.ds_map()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'type map', 'meta string:nat32 location 12 Middle Earth', 'element city 9', 'property string:nat32 name', 'property real32 x', 'property real32 y', 'element tower 3', 'property string:nat32 name', 'property real32 x', 'property real32 y', 'end_header', '8 Hobbiton 67.400002 55.5', '4 Bree 79 54.099998', '9 Rivendell 99.099998 53.299999', '5 Moria 100.5 69.5', u'11 Lothl\u00f3rien 113 74.599998', '6 Edoras 105.1 99.599998', "11 Helm's Deep 99.5 98.699997", '8 Isengard 98.5 94.800003', '12 Minas Tirith 135.60001 111.3', '7 Orthanc 98.5 94.800003', u'10 Barad-d\u00fbr 156.2 107.8', '12 Cirith Ungol 145.2 111', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_write_default(self):
before = self.ds_default()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'end_header', '']
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_default(self):
before = self.ds_default()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'end_header', '']
self.assertTrue(data=='\n'.join(expected))
temp.close()
def test_write_floats(self):
before = self.ds_floats()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'element values 3', 'property real32 x', 'property real32 y', 'end_header', '1 1.5', '2 5.5999999', '3 3.1415927', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_write_empty_element(self):
before = self.ds_empty_element()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'element dummy 0', 'end_header', '']
self.assertTrue(data=='\n'.join(expected).encode('utf8'))
temp.close()
def test_write_ints(self):
before = self.ds_ints()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'element values 3', 'property int32 x', 'end_header', '1', '2', '3', '']
self.assertTrue(data=='\n'.join(expected))
temp.close()
def test_write_colour_map(self):
before = self.ds_colour_map()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'type colour_map.rgb', 'element sample 3', 'property real32 in.r', 'property real32 in.g', 'property real32 in.b', 'property real32 out.r', 'property real32 out.g', 'property real32 out.b', 'end_header', '0 0 0 0 0 0', '1 1 1 1 1 1', '0.5 0.5 0.5 0.25 0.25 0.25', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_write_mesh(self):
before = self.ds_mesh()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = ['ply', 'format ascii 2.0', 'type mesh', 'element vertex 8', 'property real32 x', 'property real32 y', 'property real32 z', 'element face 6', 'property array:1:nat32:int32 vertex_indices', 'end_header', '1 1 -1',
'1 -1 -1', '-1 -1 -1', '-1 1 -1', '1 1 1', '-1 1 1', '-1 -1 1', '1 -1 1', '4 0 1 2 3', '4 4 5 6 7', '4 0 4 7 1', '4 1 7 6 2', '4 2 6 5 3', '4 4 0 3 5', '']
self.assertTrue(data.decode('utf8')=='\n'.join(expected))
temp.close()
def test_adv_write_read_graph(self):
for compress in ['', 'gzip', 'bzip2']:
for format in ['ascii', 'binary_little_endian', 'binary_big_endian']:
before = self.ds_graph()
before['compress'] = compress
before['format'] = format
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
after = ply2.read(temp)
temp.close()
self.equal(before, after)
def test_write_ints(self):
before = self.ds_ints()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'element values 3', 'property int32 x',
'end_header', '1', '2', '3', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_meta(self):
before = self.ds_meta()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'meta string:nat32 author 7 Cthulhu',
'meta int64 tentacles 42', 'meta real64 pi 3.141592653589793',
'end_header', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_arrays(self):
before = self.ds_arrays()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'element samples 5',
'property array:1:nat32:int8 values', 'end_header', '4 3 1 4 2',
'2 42 42', '6 100 101 102 -1 -2 0', '1 -12', '0 ', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_image(self):
before = self.ds_image()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'type image.rgb', 'element pixel 2 2',
'property nat8 red', 'property nat8 green', 'property nat8 blue',
'end_header', '0 0 0', '255 0 0', '0 255 0', '0 0 255', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_floats(self):
before = self.ds_floats()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'element values 3', 'property real32 x',
'property real32 y', 'end_header', '1 1.5', '2 5.5999999',
'3 3.1415927', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_adv_write_read_graph(self):
for compress in ['', 'gzip', 'bzip2']:
for format in [
'ascii', 'binary_little_endian', 'binary_big_endian'
]:
before = self.ds_graph()
before['compress'] = compress
before['format'] = format
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
after = ply2.read(temp)
temp.close()
self.equal(before, after)
def test_write_strings(self):
before = self.ds_strings()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'element people 8',
'property string:nat32 name', 'end_header', '9 The Alien',
'6 Ripley', '3 Ash', '7 Bite Me', '4 ', '9 Penguin',
'7 Joker ', '8 Two\nFace', ''
]
self.assertTrue(data == '\n'.join(expected))
temp.close()
def test_write_colour_map(self):
before = self.ds_colour_map()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'type colour_map.rgb',
'element sample 3', 'property real32 in.r', 'property real32 in.g',
'property real32 in.b', 'property real32 out.r',
'property real32 out.g', 'property real32 out.b', 'end_header',
'0 0 0 0 0 0', '1 1 1 1 1 1', '0.5 0.5 0.5 0.25 0.25 0.25', ''
]
self.assertTrue(data.decode('utf8') == '\n'.join(expected))
temp.close()
def test_write_mesh(self):
before = self.ds_mesh()
temp = tempfile.TemporaryFile('w+b')
ply2.write(temp, before)
temp.seek(0)
data = temp.read()
expected = [
'ply', 'format ascii 2.0', 'type mesh', 'element vertex 8',
'property real32 x', 'property real32 y', 'property real32 z',
'element face 6', 'property array:1:nat32:int32 vertex_indices',
'end_header', '1 1 -1', '1 -1 -1', '-1 -1 -1', '-1 1 -1', '1 1 1',
'-1 1 1', '-1 -1 1', '1 -1 1', '4 0 1 2 3', '4 4 5 6 7',
'4 0 4 7 1', '4 1 7 6 2', '4 2 6 5 3', '4 4 0 3 5', ''
]
self.assertTrue(data.decode('utf8') == '\n'.join(expected))
temp.close()
