def test_cube_write():
f = io.StringIO(cube_stl)
cube = read_ascii_stl(f)
cube.vertices /= 3 # rely on float reading/writing being repeatable
fo = io.StringIO("")
write_ascii_stl(cube, fo)
fo.seek(0)
cube2 = read_ascii_stl(fo)
assert is_isomorphic(cube, cube2)
def test_triange():
f = io.StringIO("""solid triangle
facet normal 0.0 0.0 -1.0
outer loop
vertex 1.0 0.0 0.0
vertex 0.0 -1.0 0.0
vertex 0.0 0.0 0.0
endloop
endfacet
endsolid triangle""")
read_ascii_stl(f)
def test_two_vertices():
f = io.StringIO("""solid triangle
facet normal 0.0 0.0 -1.0
outer loop
vertex 1.0 0.0 0.0
vertex 0.0 -1.0 0.0
endloop
endfacet
endsolid triangle""")
with pytest.raises(RuntimeError, match="Expected 'vertex'"):
read_ascii_stl(f)
def test_wrong_header():
f = io.StringIO("""soli triangle
facet normal 0.0 0.0 -1.0
outer loop
vertex 1.0 0.0 0.0
vertex 0.0 -1.0 0.0
vertex 0.0 0.0 0.0
endloop
endfacet
endsolid triangle""")
with pytest.raises(RuntimeError, match="Wrong ASCII STL header"):
read_ascii_stl(f)
def test_cube_write_binary():
f = io.StringIO(cube_stl)
cube = read_ascii_stl(f)
#cube.vertices /= 3 # won't work due to double -> float conversion
fo = io.BytesIO(b"")
write_binary_stl(cube, fo)
fo.seek(0)
cube2 = read_binary_stl(fo)
assert is_isomorphic(cube, cube2)
def test_empty():
f = io.StringIO("""solid empty
endsolid empty""")
read_ascii_stl(f)
def test_missing_endsolid():
f = io.StringIO("""solid empty
""")
with pytest.raises(RuntimeError, match="Missing 'endsolid'"):
read_ascii_stl(f)
def test_cube():
f = io.StringIO(cube_stl)
read_ascii_stl(f)