def test_write_with_normals(self):
f = StringIO()
w = ObjWriter(f)
w.write([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]],
[[0, 1, 2], [0, 1, 3]],
[[0, 0, 1], [0, 0, 1], [0, 0, 1], [0, 0, 1]],
comment="exported from test")
assert len(f.getvalue()) > 15
def test_write_with_normals(self):
f = StringIO()
w = ObjWriter(f)
w.write([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]],
[[0, 1, 2], [0, 1, 3]],
[[0, 0, 1], [0, 0, 1], [0, 0, 1], [0, 0, 1]],
comment="exported from test")
self.assertTrue(len(f.getvalue()) > 15)
def export(self, data, export_folder, project):
download_file_name = self.get_export_file_name(data)
data_file = os.path.join(export_folder, download_file_name)
with open(data_file, 'w') as f:
w = ObjWriter(f)
w.write(data.vertices, data.triangles, data.vertex_normals, comment="exported from %s" % str(data))
return download_file_name, data_file, False
def export(self, data, export_folder, project):
download_file_name = self.get_export_file_name(data)
data_file = os.path.join(export_folder, download_file_name)
with open(data_file, 'w') as f:
w = ObjWriter(f)
w.write(data.vertices,
data.triangles,
data.vertex_normals,
comment="exported from %s" % str(data))
return download_file_name, data_file, False
def test_write_parse_cycle(self):
f = StringIO()
w = ObjWriter(f)
vertices = [(0, 0, 0), (1, 0, 0), (0, 1, 0), (0, 0, 1)]
normals = [(0, 0, 1), (0, 0, 1), (0, 0, 1), (0, 0, 1)]
triangles = [(0, 1, 2), (0, 1, 3)]
w.write(vertices, triangles, normals)
f.seek(0)
p = ObjParser()
p.read(f)
assert vertices == p.vertices
assert normals == p.normals
def test_write_parse_cycle(self):
f = StringIO()
w = ObjWriter(f)
vertices = [(0, 0, 0), (1, 0, 0), (0, 1, 0), (0, 0, 1)]
normals = [(0, 0, 1), (0, 0, 1), (0, 0, 1), (0, 0, 1)]
triangles = [(0, 1, 2), (0, 1, 3)]
w.write(vertices, triangles, normals)
f.seek(0)
p = ObjParser()
p.read(f)
self.assertEqual(vertices, p.vertices)
self.assertEqual(normals, p.normals)
def test_write_simple(self):
f = StringIO()
w = ObjWriter(f)
w.write([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]],
[[0, 1, 2], [0, 1, 3]])
assert len(f.getvalue()) > 15
def test_write_simple(self):
f = StringIO()
w = ObjWriter(f)
w.write([[0, 0, 0], [1, 0, 0], [0, 1, 0], [0, 0, 1]],
[[0, 1, 2], [0, 1, 3]])
self.assertTrue(len(f.getvalue()) > 15)
