reshape(::OneToOne)

Project.toml

@@ -1,7 +1,7 @@
 name = "TensorProducts"
 uuid = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [weakdeps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"

src/onetoone.jl

@@ -5,3 +5,13 @@
 OneToOne() = OneToOne{Int}()
 Base.first(a::OneToOne) = one(eltype(a))
 Base.last(a::OneToOne) = one(eltype(a))
+
+Base.to_shape(::OneToOne) = 1
+
+# extend Base.OneTo behavior to OneToOne
+function Base.reshape(
+  parent::AbstractArray,
+  shp::Tuple{Union{Integer,Base.OneTo,OneToOne},Vararg{Union{Integer,Base.OneTo,OneToOne}}},
+)
+  return reshape(parent, Base.to_shape(shp))
+end

test/test_basics.jl

@@ -12,4 +12,8 @@ using TensorProducts: OneToOne
   @test length(a0) == 1
 
   @test blockaxes(OneToOne()) == (BlockRange(OneToOne()),)
+
+  @test reshape(ones(()), (OneToOne(),)) == ones((1,))
+  @test reshape(ones(()), (OneToOne(), Base.OneTo(1), 1)) == ones((1, 1, 1))
+  @test reshape(ones(()), OneToOne(), Base.OneTo(1), 1) == ones((1, 1, 1))
 end