Geom.to_vtk improvements
- should be integer - should have a test
This commit is contained in:
parent
a1f1b91c7f
commit
b731b1e768
|
@ -358,7 +358,7 @@ class Geom:
|
||||||
"""
|
"""
|
||||||
g = VTK.from_file(fname).geom
|
g = VTK.from_file(fname).geom
|
||||||
N_cells = g.GetNumberOfCells()
|
N_cells = g.GetNumberOfCells()
|
||||||
microstructure = np.zeros(N_cells)
|
microstructure = np.zeros(N_cells,'i')
|
||||||
grid = np.array(g.GetDimensions())-1
|
grid = np.array(g.GetDimensions())-1
|
||||||
bbox = np.array(g.GetBounds()).reshape(3,2).T
|
bbox = np.array(g.GetBounds()).reshape(3,2).T
|
||||||
size = bbox[1] - bbox[0]
|
size = bbox[1] - bbox[0]
|
||||||
|
|
|
@ -1,3 +1,6 @@
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
|
@ -65,8 +68,14 @@ class TestGeom:
|
||||||
new = Geom.from_file(f)
|
new = Geom.from_file(f)
|
||||||
assert geom_equal(new,default)
|
assert geom_equal(new,default)
|
||||||
|
|
||||||
def test_export_vtk(self,default,tmpdir):
|
def test_read_write_vtk(self,default,tmpdir):
|
||||||
default.to_vtk(str(tmpdir.join('default')))
|
default.to_vtk(str(tmpdir.join('default')))
|
||||||
|
for _ in range(3):
|
||||||
|
if os.path.exists(tmpdir.join('default.vtr')): break
|
||||||
|
time.sleep(1)
|
||||||
|
new = Geom.from_vtk(str(tmpdir.join('default.vtr')))
|
||||||
|
assert geom_equal(new,default)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('pack',[True,False])
|
@pytest.mark.parametrize('pack',[True,False])
|
||||||
def test_pack(self,default,tmpdir,pack):
|
def test_pack(self,default,tmpdir,pack):
|
||||||
|
|
Loading…
Reference in New Issue