// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

namespace Microsoft.Quantum.Samples.UnitTesting {
    open Microsoft.Quantum.Canon;
    open Microsoft.Quantum.Intrinsic;
    open Microsoft.Quantum.Diagnostics;

    /// # Summary
    /// If the Teleportation circuit is correct this operation must be an identity
    operation ApplyIdentityWithTeleport(arg : Qubit[]) : Unit {
        EqualityFactI(Length(arg), 1, "Helper is defined only on single qubit input");

        use auxiliary = Qubit();
        RunTeleportation(arg[0], auxiliary);
        SWAP(arg[0], auxiliary);
    }

    /// # Summary
    /// Tests the correctness of the teleportation circuit from Teleportation.qs
    @Test("QuantumSimulator")
    operation CheckTeleportIdentityIsNoOp() : Unit {
        // given that there is randomness involved in the Teleportation,
        // repeat the tests several times.
        for idxIteration in 1 .. 8 {
            for assertion in [AssertOperationsEqualInPlace, AssertOperationsEqualReferenced] {
                assertion(1, ApplyIdentityWithTeleport, NoOp<Qubit[]>);
            }
        }
    }

}