Boson Beach Mathematicians at the resort.py

import json
import pennylane as qml
import pennylane.numpy as np

dev = qml.device("default.qubit", wires = 5)

@qml.qnode(dev)
def circuit():
    """
    Circuit in which you will prepare the Bell state with the allowed gates.
    """


    # Put your code here #
    qml.Hadamard(wires=[0])

    qml.QFT(wires=[2, 1, 0])
    qml.QFT(wires=[2, 1, 0])
    qml.QFT(wires=[4, 3, 2])
    qml.QFT(wires=[4, 3, 2]) 

    return qml.probs(wires = range(5))


# These functions are responsible for testing the solution

def run(case: str) -> str:
    return "No output"

def check(have: str, want: str) -> None:

    assert np.isclose(circuit()[0], 0.5), "The state is not correct"
    assert np.isclose(circuit()[-1], 0.5), "The state is not correct"

    for op in circuit.tape.operations:
      assert (isinstance(op, qml.Hadamard) or isinstance(op, qml.T) or isinstance(op, qml.QFT)), f"You can only use Hadamard, T and QFT operators. You are using {op.name}"
      if isinstance(op, qml.QFT):
        assert len(op.wires) == 3, "QFT must act on 3 wires"


# These are the public test cases
test_cases = [
    ('No input', 'No output')
]

# This will run the public test cases locally
for i, (input_, expected_output) in enumerate(test_cases):
    print(f"Running test case {i} with input '{input_}'...")

    try:
        output = run(input_)

    except Exception as exc:
        print(f"Runtime Error. {exc}")

    else:
        if message := check(output, expected_output):
            print(f"Wrong Answer. Have: '{output}'. Want: '{expected_output}'.")

        else:
            print("Correct!")