ising_gates_test.py

import cirq
import pytest

import ising_gates as ig
from ising_circuit import IsingCircuit

# @Source This is based on the pytests from the homeworks earlier in the semester

# This will test the shift gates on n qubits
@pytest.mark.parametrize('n_qubits', range(2, 10))
def test_shiftu(n_qubits):
    # Makes a circuit, applies a hadamard to the first qubit, and shifts up
    qubits1 = cirq.LineQubit.range(n_qubits)
    qubits2 = cirq.LineQubit.range(n_qubits)
    circuit1 = cirq.Circuit()
    circuit1.append([cirq.H(qubits1[0])])
    circuit1.append(ig.SHIFTU(n_qubits).on(*qubits1))
    simulator1 = cirq.Simulator()
    result1 = simulator1.simulate(circuit1, qubit_order=qubits1)
    # Second circuit that should match the end state, |+> tensored with |1> * n
    circuit2 = cirq.Circuit()
    circuit2.append([cirq.H(qubits2[0])])
    for q in qubits2[1:n_qubits]:
        circuit2.append([cirq.X(q)])
    simulator2 = cirq.Simulator()
    result2 = simulator2.simulate(circuit2, qubit_order=qubits2)
    assert (result1.final_state == result2.final_state).all()


@pytest.mark.parametrize('n_qubits', range(2, 10))
def test_shiftd(n_qubits):
    # Makes a circuit, applies a hadamard to the first qubit, shifts up, then
    # shifts down. This should show that shiftd == shiftu**-1
    qubits = cirq.LineQubit.range(n_qubits)
    circuit = cirq.Circuit()
    circuit.append([cirq.H(qubits[0])])
    simulator = cirq.Simulator()
    origin = simulator.simulate(circuit, qubit_order=qubits)
    circuit.append(ig.SHIFTU(n_qubits).on(*qubits))
    circuit.append(ig.SHIFTD(n_qubits).on(*qubits))
    result = simulator.simulate(circuit, qubit_order=qubits)

    assert (origin.final_state == result.final_state).all()


@pytest.mark.parametrize('n_qubits', [4, 8, 16, 32])
def test_printing(n_qubits):
    ic = IsingCircuit(n_qubits, 0.1, 2, 200, 10)
    assert isinstance(ic.circuit.to_text_diagram(), str)