Generalize axes type in similar (#7)

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "BroadcastMapConversion"
 uuid = "4a4adec5-520f-4750-bb37-d5e66b4ddeb2"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [deps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"

src/BroadcastMapConversion.jl

@@ -165,7 +165,7 @@ function mapped(f, args...)
 end
 
 Base.similar(m::Mapped, elt::Type) = similar(Broadcasted(m), elt)
-Base.similar(m::Mapped, elt::Type, ax::Tuple) = similar(Broadcasted(m), elt, ax)
+Base.similar(m::Mapped, elt::Type, ax) = similar(Broadcasted(m), elt, ax)
 Base.axes(m::Mapped) = axes(Broadcasted(m))
 # Equivalent to:
 # map(m.f, m.args...)

test/test_basics.jl

@@ -21,6 +21,9 @@ using Test: @inferred, @test, @test_throws, @testset
     @test axes(m′) == axes(bc)
     @test copyto!(similar(m, elt), m) ≈ ref
     @test copyto!(similar(m′, elt), m) ≈ ref
+    x = similar(m, Float32, Base.OneTo.((3, 2)))
+    @test x isa Matrix{Float32}
+    @test size(x) == (3, 2)
   end
 
   @test @inferred is_map_expr(