Better fix, add tests

Author: mtfishman

src/abstractsparsearrayinterface.jl

@@ -1,3 +1,4 @@
+using Base: @_propagate_inbounds_meta
 using DerivableInterfaces: DerivableInterfaces, @derive, @interface, AbstractArrayInterface
 
 # This is to bring `ArrayLayouts.zero!` into the namespace
@@ -37,13 +38,19 @@ end
 @interface ::AbstractArrayInterface function isstored(
   a::AbstractArray{<:Any,N}, I::Vararg{Int,N}
 ) where {N}
+  @_propagate_inbounds_meta
   @boundscheck checkbounds(a, I...)
   return true
 end
-@interface ::AbstractArrayInterface function isstored(a::AbstractArray, I::Int...)
+@interface interface::AbstractArrayInterface function isstored(a::AbstractArray, I::Int)
+  @_propagate_inbounds_meta
+  return @interface interface isstored(a, Tuple(CartesianIndices(a)[I])...)
+end
+@interface interface::AbstractArrayInterface function isstored(a::AbstractArray, I::Int...)
+  @_propagate_inbounds_meta
   @boundscheck checkbounds(a, I...)
   I′ = ntuple(i -> I[i], ndims(a))
-  return isstored(a, I′...)
+  return @inbounds @interface interface isstored(a, I′...)
 end
 @interface ::AbstractArrayInterface eachstoredindex(a::AbstractArray) = eachindex(a)
 @interface ::AbstractArrayInterface getstoredindex(a::AbstractArray, I::Int...) =
@@ -163,6 +170,12 @@ function Base.setindex!(a::StoredValues, value, I::Int)
   return setstoredindex!(a.array, value, a.storedindices[I])
 end
 
+@interface ::AbstractSparseArrayInterface function isstored(
+  a::AbstractArray{<:Any,N}, I::Vararg{Int,N}
+) where {N}
+  return CartesianIndex(I) in eachstoredindex(a)
+end
+
 @interface ::AbstractSparseArrayInterface storedvalues(a::AbstractArray) = StoredValues(a)
 
 @interface ::AbstractSparseArrayInterface function eachstoredindex(

src/sparsearraydok.jl

@@ -73,8 +73,8 @@ storage(a::SparseArrayDOK) = a.storage
 Base.size(a::SparseArrayDOK) = a.size
 
 storedvalues(a::SparseArrayDOK) = values(storage(a))
-function isstored(a::SparseArrayDOK{<:Any,N}, I::Vararg{Int,N}) where {N}
-  return CartesianIndex(I) in keys(storage(a))
+function isstored(a::SparseArrayDOK, I::Int...)
+  return @interface interface(a) isstored(a, I...)
 end
 function eachstoredindex(a::SparseArrayDOK)
   return keys(storage(a))

test/test_basics.jl

@@ -26,6 +26,9 @@ arrayts = (Array, JLArray)
   for indexstyle in (IndexLinear(), IndexCartesian())
     for I in eachindex(indexstyle, a)
       @test isstored(a, I)
+      if indexstyle == IndexCartesian()
+        @test isstored(a, Tuple(I)..., 1)
+      end
     end
   end
   @test eachstoredindex(a) == eachindex(a)

test/test_sparsearraydok.jl

@@ -210,6 +210,10 @@ arrayts = (Array,)
     a[1, 2] = 12
     @test sprint(show, "text/plain", a) ==
       "$(summary(a)):\n  ⋅   $(eltype(a)(12))\n  ⋅     ⋅ "
+
+    a = SparseArrayDOK{elt}(undef, 2)
+    a[1] = 1
+    @test sprint(show, "text/plain", a) == "$(summary(a)):\n $(eltype(a)(1))\n  ⋅ "
   end
 
   # Regression test for: