@@ -77,25 +77,25 @@ def uvw_fesom_channel(ds_fesom_channel) -> VectorField:
7777def test_fesom_fieldset (ds_fesom_channel , uv_fesom_channel ):
7878 fieldset = FieldSet ([uv_fesom_channel , uv_fesom_channel .U , uv_fesom_channel .V ])
7979 # Check that the fieldset has the expected properties
80- assert (fieldset .U == ds_fesom_channel .U ).all ()
81- assert (fieldset .V == ds_fesom_channel .V ).all ()
80+ assert (fieldset .U . data == ds_fesom_channel .U ).all ()
81+ assert (fieldset .V . data == ds_fesom_channel .V ).all ()
8282
8383
8484def test_fesom_in_particleset (ds_fesom_channel , uv_fesom_channel ):
8585 fieldset = FieldSet ([uv_fesom_channel , uv_fesom_channel .U , uv_fesom_channel .V ])
8686
8787 # Check that the fieldset has the expected properties
88- assert (fieldset .U == ds_fesom_channel .U ).all ()
89- assert (fieldset .V == ds_fesom_channel .V ).all ()
88+ assert (fieldset .U . data == ds_fesom_channel .U ).all ()
89+ assert (fieldset .V . data == ds_fesom_channel .V ).all ()
9090 pset = ParticleSet (fieldset , pclass = Particle )
9191 assert pset .fieldset == fieldset
9292
9393
9494def test_set_interp_methods (ds_fesom_channel , uv_fesom_channel ):
9595 fieldset = FieldSet ([uv_fesom_channel , uv_fesom_channel .U , uv_fesom_channel .V ])
9696 # Check that the fieldset has the expected properties
97- assert (fieldset .U == ds_fesom_channel .U ).all ()
98- assert (fieldset .V == ds_fesom_channel .V ).all ()
97+ assert (fieldset .U . data == ds_fesom_channel .U ).all ()
98+ assert (fieldset .V . data == ds_fesom_channel .V ).all ()
9999
100100 # Set the interpolation method for each field
101101 fieldset .U .interp_method = UXPiecewiseConstantFace
0 commit comments