update measure

QuantumBFS · Apr 25, 2019 · adaab56 · adaab56
1 parent 8d9cb01
commit adaab56
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/register.jl b/src/register.jl
@@ -143,10 +143,10 @@ function YaoBase.addbits!(r::ArrayReg, n::Int)
     return r
 end
 
-function YaoBase.insert_qubits!(reg::ArrayReg{B}, loc::Int; nbit::Int=1) where B
+function YaoBase.insert_qubits!(reg::ArrayReg{B}, loc::Int; nqubits::Int=1) where B
     na = nactive(reg)
     focus!(reg, 1:loc-1)
-    reg2 = join(zero_state(nbit, B), reg) |> relax! |> focus!((1:na+nbit)...)
+    reg2 = join(zero_state(nqubits; nbatch=B), reg) |> relax! |> focus!((1:na+nqubits)...)
     reg.state = reg2.state
     reg
 end

diff --git a/test/focus.jl b/test/focus.jl
@@ -35,6 +35,8 @@ end
     @test copy(reg) |> addbits!(2) |> nactive == 5
     reg2 = copy(reg) |> addbits!(2) |> focus!(4,5)
     @test (reg2 |> measure_remove!; reg2) |> relax!(to_nactive=nqubits(reg2)) ≈ reg
+
+    @test insert_qubits!(copy(reg), 2; nqubits=2) |> nactive == 5
 end
 
 @testset "Focus 2" begin