[WIP] Fix vector show

Author: mtfishman

src/abstractblocksparsearray/wrappedabstractblocksparsearray.jl

@@ -23,6 +23,12 @@ const AnyAbstractBlockSparseArray{T,N} = Union{
   <:AbstractBlockSparseArray{T,N},<:WrappedAbstractBlockSparseArray{T,N}
 }
 
+const AnyAbstractBlockSparseVector{T} = AnyAbstractBlockSparseArray{T,1}
+const AnyAbstractBlockSparseMatrix{T} = AnyAbstractBlockSparseArray{T,2}
+const AnyAbstractBlockSparseVecOrMat{T,N} = Union{
+  AnyAbstractBlockSparseVector{T},AnyAbstractBlockSparseMatrix{T}
+}
+
 function DerivableInterfaces.interface(::Type{<:AnyAbstractBlockSparseArray})
   return BlockSparseArrayInterface()
 end
@@ -355,9 +361,9 @@ function SparseArraysBase.isstored(
 end
 
 function Base.replace_in_print_matrix(
-  A::AnyAbstractBlockSparseArray{<:Any,2}, i::Integer, j::Integer, s::AbstractString
+  a::AnyAbstractBlockSparseVecOrMat, i::Integer, j::Integer, s::AbstractString
 )
-  return isstored(A, i, j) ? s : Base.replace_with_centered_mark(s)
+  return isstored(a, i, j) ? s : Base.replace_with_centered_mark(s)
 end
 
 # attempt to catch things that wrap GPU arrays

src/blocksparsearray/blocksparsearray.jl

@@ -174,6 +174,12 @@ function BlockSparseArray{T,N}(
   return BlockSparseArray{T,N,Array{T,N}}(undef, axes)
 end
 
+function BlockSparseArray{T,N}(
+  ::UndefInitializer, axes::Tuple{Vararg{AbstractUnitRange{<:Integer}}}
+) where {T,N}
+  return throw(ArgumentError("Length of axes doesn't match number of dimensions."))
+end
+
 function BlockSparseArray{T,N}(
   ::UndefInitializer, axes::Vararg{AbstractUnitRange{<:Integer},N}
 ) where {T,N}

test/test_basics.jl

@@ -130,6 +130,15 @@ arrayts = (Array, JLArray)
         @test iszero(storedlength(a))
       end
     end
+
+    for dims in (
+      ([2, 2], [2, 2]),
+      (([2, 2], [2, 2]),),
+      blockedrange.(([2, 2], [2, 2])),
+      (blockedrange.(([2, 2], [2, 2])),),
+    )
+      @test_throws ArgumentError BlockSparseVector{elt}(undef, dims...)
+    end
   end
   @testset "blockstype, blocktype" begin
     a = arrayt(randn(elt, 2, 2))