Ensure (ux)fieldset has both UV and UVW for datasets with U, V, and W

Author: fluidnumericsJoe

src/parcels/_core/fieldset.py

@@ -200,6 +200,9 @@ def from_ugrid_conventions(cls, ds: ux.UxDataset, mesh: str = "spherical"):
                 fields["UVW"] = VectorField(
                     "UVW", fields["U"], fields["V"], fields["W"], vector_interp_method=Ux_Velocity
                 )
+                fields["UV"] = VectorField(
+                    "UV", fields["U"], fields["V"], vector_interp_method=Ux_Velocity
+                )
             else:
                 fields["UV"] = VectorField("UV", fields["U"], fields["V"], vector_interp_method=Ux_Velocity)
 

tests/test_fieldset.py

@@ -239,6 +239,7 @@ def test_fieldset_from_fesom2():
     fieldset = FieldSet.from_ugrid_conventions(ds)
     assert "U" in fieldset.fields
     assert "V" in fieldset.fields
+    assert "UV" in fieldset.fields
     assert "UVW" in fieldset.fields