fix QuantumBFS/YaoBlocks.jl#21

QuantumBFS · Apr 24, 2019 · 51d5323 · 51d5323
1 parent b4ec7c9
commit 51d5323
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/register.jl b/src/register.jl
@@ -114,7 +114,7 @@ ArrayReg(r::ArrayReg{B}) where B = ArrayReg{B}(copy(r.state))
 Base.copy(r::ArrayReg) = ArrayReg(r)
 Base.similar(r::ArrayRegOrAdjointArrayReg{B}) where B = ArrayReg{B}(similar(state(r)))
 
-function Base.copyto!(dst::R, src::R) where {R <: ArrayRegOrAdjointArrayReg}
+function Base.copyto!(dst::ArrayRegOrAdjointArrayReg, src::ArrayRegOrAdjointArrayReg)
     copyto!(state(dst), state(src))
     return dst
 end

diff --git a/test/register.jl b/test/register.jl
@@ -116,3 +116,11 @@ end
     focus!(r, 2:4)
     @test sum(r.state, dims=2) ≈ ArrayReg(bit"011").state
 end
+
+@testset "YaoBlocks.jl/issues/21" begin
+    st = rand(ComplexF64, 16, 2)
+    r1 = ArrayReg(view(st, :, 1))
+    r2 = ArrayReg(rand(ComplexF64, 16, 1))
+    copyto!(r1, r2)
+    @test r1 == r2
+end