## Enter Team ID 
import os 
os.environ["TEAMID"] = "Excalibur"

from qiskit import QuantumCircuit

def make_bell_state():
    qc = QuantumCircuit(2)
    ### your code here
    qc.x(0)
    qc.h(0)
    qc.cx(0,1)
    ### your code here
    return qc 

def test_function_1():
    circuit = make_bell_state()
    return circuit 

test_function_1().draw()
    

from grader.graders.problem_1.grader import grader1 
grader1.evaluate(make_bell_state)

def superposition_operation(n):
    qc = QuantumCircuit(n)
    
    ### Your code here
    for i in range(n):
        qc.h(i)
    ### Your code here
    
    return qc

def test_function_2():
    n = 5

    operation = superposition_operation(n)
    
    return operation 
test_function_2().draw()
    

from grader.graders.problem_1.grader import grader2
grader2.evaluate(superposition_operation)

def make_even_odd(n):
    even = QuantumCircuit(n)
    odd = QuantumCircuit(n)
    
    ### your code here 
    for i in range(1,n):
        even.h(i)
        odd.h(i)
    odd.x(0)
    ### your code here 
    
    return (even, odd)

def test_function_3():
    n = 3 
    
    even, odd = make_even_odd(n)
    
    return even, odd 

even, odd = test_function_3()

display(even.draw('mpl'))
odd.draw('mpl')

from grader.graders.problem_1.grader import grader3
grader3.evaluate(make_even_odd)

