add tests for removefield; allow specifying an Tuple of fields to be removed

Author: boriskaus

src/utils.jl

@@ -120,7 +120,6 @@ function removefield(V::AbstractGeneralGrid,field_name::Symbol)
     return V
 end
 
-
 """
     V = removefield(V::AbstractGeneralGrid,field_name::String)
 
@@ -131,6 +130,21 @@ function removefield(V::AbstractGeneralGrid,field_name::String)
     return removefield(V,Symbol(field_name))
 end
 
+"""
+    V = removefield(V::AbstractGeneralGrid,field_name::NTuple{N,Symbol})
+
+Removes the fields in the tuple `field_name` from the GeoData or CartData dataset
+
+"""
+function removefield(V::AbstractGeneralGrid,field_name::NTuple{N,Symbol})   where N
+    
+    for ifield=1:N
+        V = removefield(V,field_name[ifield])
+    end
+
+    return V
+end
+
 
 """
 cross_section_volume(Volume::AbstractGeneralGrid; dims=(100,100), Interpolate=false, Depth_level=nothing; Lat_level=nothing; Lon_level=nothing; Start=nothing, End=nothing, Depth_extent=nothing )
@@ -1831,3 +1845,6 @@ function inpoly_fast(PolyX::Vector{T}, PolyY::Vector{T}, x::T, y::T) where T <:
     end
     return inside
 end
+
+
+

test/test_utils.jl

@@ -44,6 +44,16 @@ Data_set3D      =   addfield(Data_set3D,"Lat", Lat)
 Data_set3D      =   addfield(Data_set3D,(;Lat, Lon))
 @test keys(Data_set3D.fields) == (:Depthdata, :LonData, :Velocity, :Lat, :Lon)
 
+# test removefield
+Data_set3D_1    =   removefield(Data_set3D,"Lon")
+@test keys(Data_set3D_1.fields) == (:Depthdata, :LonData, :Velocity, :Lat)
+
+Data_set3D_2    =   removefield(Data_set3D,:Lon)
+@test keys(Data_set3D_2.fields) == (:Depthdata, :LonData, :Velocity, :Lat)
+
+Data_set3D_3    =   removefield(Data_set3D,(:Lon,:Lat))
+@test keys(Data_set3D_3.fields) == (:Depthdata, :LonData, :Velocity)
+
 # Create 3D cartesian dataset
 Data_setCart3D  =   CartData(Lon,Lat,Depth,(Depthdata=Data,LonData=Lon, Velocity=(Vx,Vy,Vz)))