abp/tests/test_get_state_vector.py

from abp.graphstate import GraphState
from abp import qi
import numpy as np

def test_single_qubit():
    """ Test some single-qubit stuff """
    g = GraphState()
    g.add_node(0)
    g.add_node(1)
    g.act_local_rotation_by_name(0, "hadamard")
    g.act_local_rotation_by_name(1, "hadamard")
    g.act_cz(0, 1)
    assert np.allclose(g.to_state_vector().state, qi.bond)