From c5f286512155bac7aefe5750f642c0d8becb0b3f Mon Sep 17 00:00:00 2001 From: Philip Eisenlohr Date: Tue, 15 Feb 2022 18:04:21 -0500 Subject: [PATCH] make apparent that vtk.get automatically "squeezes" --- python/tests/test_VTK.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/tests/test_VTK.py b/python/tests/test_VTK.py index be3e8d6f3..69119ff12 100644 --- a/python/tests/test_VTK.py +++ b/python/tests/test_VTK.py @@ -164,8 +164,7 @@ class TestVTK: 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)) for k,s in shapes.items(): - assert np.allclose(default.get(k).reshape((N,-1)),d[k]['data'], - rtol=1e-7) + assert np.allclose(np.squeeze(d[k]['data']),default.get(k),rtol=1e-7) def test_add_masked(self,default):