diff --git a/tests/commit/field/test__angular_velocity.py b/tests/commit/field/test__angular_velocity.py index d6141c1e5..51d0d8dc6 100644 --- a/tests/commit/field/test__angular_velocity.py +++ b/tests/commit/field/test__angular_velocity.py @@ -8,6 +8,6 @@ class TestAngularVelocity(TestCase): def test_sample_at(self): field = AngularVelocity(location=vec(x=0, y=0)) - self.assertEqual(channel(vector='x,y'), field.shape.channel) field @ CenteredGrid(0, x=4, y=3) + self.assertEqual(channel(vector='x,y'), (field @ CenteredGrid(0, x=4, y=3)).shape.channel) field @ StaggeredGrid(0, x=4, y=3)