make apparent that vtk.get automatically "squeezes"

This commit is contained in:
Philip Eisenlohr 2022-02-15 18:04:21 -05:00
parent a8898163ce
commit c5f2865121
1 changed files with 1 additions and 2 deletions

View File

@ -164,8 +164,7 @@ class TestVTK:
data = np.random.random(N*np.prod(s)).reshape((N,-1))) data = np.random.random(N*np.prod(s)).reshape((N,-1)))
default.add(Table(np.column_stack([d[k]['data'] for k in shapes.keys()]),shapes)) default.add(Table(np.column_stack([d[k]['data'] for k in shapes.keys()]),shapes))
for k,s in shapes.items(): for k,s in shapes.items():
assert np.allclose(default.get(k).reshape((N,-1)),d[k]['data'], assert np.allclose(np.squeeze(d[k]['data']),default.get(k),rtol=1e-7)
rtol=1e-7)
def test_add_masked(self,default): def test_add_masked(self,default):