fix lookup for related variables when indexing with a scalar

JuliaGeo · Dec 8, 2023 · 3209c79 · 3209c79
1 parent 7ca6b69
commit 3209c79
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/src/subvariable.jl b/src/subvariable.jl
@@ -175,7 +175,7 @@ groupname(ds::SubDataset) = groupname(ds.ds)
 
 
 function dataset(v::SubVariable)
-    indices = (;((Symbol(d),i) for (d,i) in zip(dimnames(v),v.indices))...)
+    indices = (;((Symbol(d),i) for (d,i) in zip(dimnames(v.parent),v.indices))...)
     return SubDataset(dataset(v.parent),indices)
 end
 

diff --git a/test/test_attrib.jl b/test/test_attrib.jl
@@ -111,7 +111,6 @@ defVar(ds,:temperature2,data,(:lon,:lat),
        attrib = OrderedDict(:long_name => "Temperature",
                  :test_vector_attrib => [1,2,3]))
 
-
 defVar(ds,:temperature3,data,(:lon,:lat),
        attrib = Dict(:long_name => "Temperature",
                  :test_vector_attrib => [1,2,3]))

diff --git a/test/test_subvariable.jl b/test/test_subvariable.jl
@@ -195,3 +195,22 @@ close(ds_slice)
 @test TDS(fname_slice)["lon"][:] == 2:3
 
 close(ds)
+
+
+
+#
+
+fname = tempname()
+ds = TDS(fname,"c")
+
+nclon = defVar(ds,"lon", 1:7, ("lon",))
+nclat = defVar(ds,"lat", 1:10, ("lat",))
+nctime = defVar(ds,"time", [DateTime(2000,1,1)], ("time",))
+ncsst = defVar(ds,"sst", ones(7,10,1), ("lon", "lat", "time"))
+
+ncsst3 = @select(ncsst,lon ≈ 2 && lat ≈ 3)
+
+ss = ncsst3["time"]
+@test ss[1] == DateTime(2000,1,1)
+
+close(ds)