Fix 155 (#156)

JuliaIO · Feb 23, 2024 · 24bbb8e · 24bbb8e · meggart · Feb 29, 2024
1 parent 387d910
commit 24bbb8e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/diskarray.jl b/src/diskarray.jl
@@ -153,6 +153,7 @@ splitcs(::Tuple{}, csnow, csrem) = (csnow, csrem)
 
 function getindex_disk(a, i::Union{Integer,CartesianIndex}...)
     checkscalar(i)
+    checkbounds(a,i...)
     outputarray = Array{eltype(a)}(undef, map(_ -> 1, size(a))...)
     i = Base.to_indices(a, i)
     j = map(1:ndims(a)) do d

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -28,6 +28,12 @@ end
     @test DiskArrays.can_scalar() == true
     @test DiskArrays.checkscalar(Bool, 1, 2, 3) == true
     @test DiskArrays.checkscalar(Bool, :, 2:5, 3) == true
+    a = AccessCountDiskArray(reshape(1:24,2,3,4),chunksize=(2,2,2))
+    @test a[1,2,3] == 15
+    @test a[1,2,3,1] == 15
+    @test_throws BoundsError a[1,2]
+    @test a[CartesianIndex(1,2),3] == 15
+    @test a[CartesianIndex(1,2,3)] == 15
 end
 
 function test_getindex(a)