Fix similar with eltype change (#44)

Author: ogauthe

Project.toml

@@ -1,7 +1,7 @@
 name = "BlockSparseArrays"
 uuid = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.2.16"
+version = "0.2.17"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

ext/BlockSparseArraysGradedUnitRangesExt/BlockSparseArraysGradedUnitRangesExt.jl

@@ -2,7 +2,7 @@ module BlockSparseArraysGradedUnitRangesExt
 
 using BlockSparseArrays: AnyAbstractBlockSparseArray, BlockSparseArray, blocktype
 using GradedUnitRanges: AbstractGradedUnitRange
-using TypeParameterAccessors: set_ndims, unwrap_array_type
+using TypeParameterAccessors: set_eltype, set_ndims, unwrap_array_type
 
 # A block spare array similar to the input (dense) array.
 # TODO: Make `BlockSparseArrays.blocksparse_similar` more general and use that,
@@ -15,7 +15,9 @@ function similar_blocksparse(
   # TODO: Probably need to unwrap the type of `a` in certain cases
   # to make a proper block type.
   return BlockSparseArray{
-    elt,length(axes),set_ndims(unwrap_array_type(blocktype(a)), length(axes))
+    elt,
+    length(axes),
+    set_eltype(set_ndims(unwrap_array_type(blocktype(a)), length(axes)), elt),
   }(
     axes
   )

test/test_gradedunitrangesext.jl

@@ -53,6 +53,9 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
       @test 2 * Array(a) == b
     end
 
+    b = similar(a, ComplexF64)
+    @test eltype(b) === ComplexF64
+
     # Test mixing graded axes and dense axes
     # in addition/broadcasting.
     for b in (a + Array(a), Array(a) + a)