Skip to content

Commit ddd98d6

Browse files
committed
add tests for open_mfdataset
1 parent e13e918 commit ddd98d6

File tree

2 files changed

+34
-1
lines changed

2 files changed

+34
-1
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "YAXArrays"
22
uuid = "c21b50f5-aa40-41ea-b809-c0f5e47bfa5c"
33
authors = ["Fabian Gans <[email protected]>"]
4-
version = "0.5.14"
4+
version = "0.5.15"
55

66
[deps]
77
CFTime = "179af706-886a-5703-950a-314cd64e0468"

test/Datasets/datasets.jl

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -531,3 +531,36 @@ end
531531
ds = Dataset(a=a1, b=a2, c=a3, d=a4, e=a5)
532532
@test_throws ArgumentError Cube(ds)
533533
end
534+
535+
@testset "Open_mfdataset" begin
536+
import NetCDF
537+
d1 = DD.X(1:20)
538+
d2 = DD.Y(1:10)
539+
a1 = rand(20,10)
540+
a2 = rand(20,10)
541+
td = mktempdir()
542+
f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
543+
544+
array1 = YAXArray((d1,d2),a1)
545+
array2 = YAXArray((d1,d2),a2)
546+
547+
savecube(array1,f1)
548+
savecube(array2,f2)
549+
550+
ds = open_mfdataset(DD.DimArray([f1,f2],(DD.Ti(1:2),)))
551+
552+
@test ds.layer.data[:,:,1] == array1
553+
@test ds.layer.data[:,:,2] == array2
554+
555+
td = mktempdir()
556+
f1, f2 = joinpath.(td,("file_1.nc","file_2.nc"))
557+
td1, td2 = DD.Ti(1:2), DD.Ti(3:4)
558+
a1,a2 = rand(20,10,2), rand(20,10,2)
559+
array1,array2 = YAXArray((d1,d2,td1),a1), YAXArray((d1,d2,td2),a2)
560+
savecube(array1,f1)
561+
savecube(array2,f2)
562+
563+
ds = open_mfdataset(joinpath(td,"*.nc"))
564+
@test ds.layer.data[:,:,1:2] == array1
565+
@test ds.layer.data[:,:,3:4] == array2
566+
end

0 commit comments

Comments
 (0)