reshape(::OneToOne) (#10)

Author: ogauthe

Project.toml

@@ -1,7 +1,7 @@
 name = "TensorProducts"
 uuid = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 authors = ["ITensor developers <support@itensor.org> and contributors"]
-version = "0.1.5"
+version = "0.1.6"
 
 [weakdeps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"

src/onetoone.jl

@@ -5,3 +5,6 @@ struct OneToOne{T} <: AbstractUnitRange{T} end
 OneToOne() = OneToOne{Int}()
 Base.first(a::OneToOne) = one(eltype(a))
 Base.last(a::OneToOne) = one(eltype(a))
+
+# impose Int64 to keep Base.to_shape(::Base.OneTo) convention
+Base.to_shape(::OneToOne) = 1

test/test_basics.jl

@@ -12,4 +12,6 @@ using TensorProducts: OneToOne
   @test length(a0) == 1
 
   @test blockaxes(OneToOne()) == (BlockRange(OneToOne()),)
+  @test Base.to_shape(OneToOne()) isa Int64
+  @test Base.to_shape(OneToOne()) == 1
 end