|
- from abp.graphstate import GraphState
- from anders_briegel import graphsim
- import random
- import difflib
- import re
-
- def compare(a, b):
- """ Sketchy as you like. Remove this abomination """
- aa = a.get_adj_list()
- bb = b.adj_list()
- try:
- assert re.sub("\\s", "", aa) == re.sub("\\s", "", bb)
- except AssertionError:
- print aa
- print bb
- raise
-
- def test_hadamard():
- """ Test hadamards """
- a = graphsim.GraphRegister(1)
- b = GraphState()
- b.add_node(0)
-
- compare(a, b)
- a.hadamard(0)
- b.act_hadamard(0)
- compare(a, b)
- a.hadamard(0)
- b.act_hadamard(0)
- compare(a, b)
-
-
- def test_local_rotations():
- """ Test local rotations """
- a = graphsim.GraphRegister(1)
- b = GraphState()
- b.add_node(0)
- compare(a, b)
-
- for i in range(1000):
- j = random.randint(0, 23)
- a.local_op(0, graphsim.LocCliffOp(j))
- b.act_local_rotation(0, j)
- compare(a, b)
-
-
- def test_cz_table():
- """ Test the CZ table """
- for j in range(24):
- a = graphsim.GraphRegister(2)
- b = GraphState()
- b.add_node(0)
- b.add_node(1)
- compare(a, b)
-
- a.local_op(0, graphsim.LocCliffOp(j))
- b.act_local_rotation(0, j)
-
- a.local_op(1, graphsim.LocCliffOp(j))
- b.act_local_rotation(1, j)
-
- a.cphase(0, 1)
- b.act_cz(0, 1)
- compare(a, b)
-
-
-
- def test_1():
- """ TODO: this one always succeeds """
- N=10
-
- a = graphsim.GraphRegister(N)
- b = GraphState()
-
- for i in range(N):
- a.hadamard(i)
- b.add_node(i)
- b.act_hadamard(i)
-
- for i in range(N-1):
- a.cphase(i, i+1)
- b.act_cz(i, i+1)
-
- compare(a, b)
-
-
- def _test_2():
- """ TODO: This one fails at the moment """
- N=10
-
- a = graphsim.GraphRegister(N)
- b = GraphState()
-
- for i in range(N):
- b.add_node(i)
-
- for i in range(100):
- if random.random()>0.5:
- j = random.randint(0, N-1)
- a.hadamard(j)
- b.act_hadamard(j)
- else:
- q = random.randint(0, N-2)
- a.cphase(q, q+1)
- b.act_cz(q, q+1)
-
- aa = a.get_adj_list()
- bb = b.adj_list()
- compare(a, b)
|