From ea0033379cf0fe08a08af1c04c74e2542fe3f5e9 Mon Sep 17 00:00:00 2001
From: Pete Shadbolt <pete.shadbolt@gmail.com>
Date: Wed, 11 May 2016 02:26:30 +0100
Subject: [PATCH] QTIP, not much

---
 tests/test_against_circuit_model.py | 46 ++++++++++++++++++-----------
 tests/test_qi_vs_qtip.py            |  9 ++++++
 2 files changed, 37 insertions(+), 18 deletions(-)
 create mode 100644 tests/test_qi_vs_qtip.py

diff --git a/tests/test_against_circuit_model.py b/tests/test_against_circuit_model.py
index 2c9d4d7..b18a627 100644
--- a/tests/test_against_circuit_model.py
+++ b/tests/test_against_circuit_model.py
@@ -29,21 +29,31 @@ def test_hadamard_only_multiqubit(n=6):
 
 def test_all_multiqubit(n=4):
     """ A multi qubit test with arbitrary local rotations """
-    for qqq in range(REPEATS):
-        g = GraphState(range(n))
-        c = CircuitModel(n)
-        for i in range(10):
-            qubit = np.random.randint(0, n - 1)
-            rotation = np.random.randint(0, 24 - 1)
-            g.act_local_rotation(qubit, rotation)
-            c.act_local_rotation(qubit, clifford.unitaries[rotation])
-
-        assert g.to_state_vector() == c
-
-        for i in range(1):
-            a, b = np.random.randint(0, n-1, 2)
-            if a != b:
-                g.act_cz(a, b)
-                c.act_cz(a, b)
-
-        assert g.to_state_vector() == c
+    g = GraphState(range(n))
+    c = CircuitModel(n)
+    for i in range(10):
+        qubit = np.random.randint(0, n - 1)
+        rotation = np.random.randint(0, 24 - 1)
+        g.act_local_rotation(qubit, rotation)
+        c.act_local_rotation(qubit, clifford.unitaries[rotation])
+
+    assert g.to_state_vector() == c
+
+    for i in range(100):
+        a, b = np.random.randint(0, n-1, 2)
+        if a != b:
+            g.act_cz(a, b)
+            c.act_cz(a, b)
+            assert np.allclose(np.sum(np.abs(c.state)**2), 1)
+            assert np.allclose(np.sum(np.abs(g.to_state_vector().state)**2), 1)
+
+            if not g.to_state_vector() == c:
+                print g
+                print a, b
+                print "Circuit:"
+                print g.to_state_vector()
+                print "Graph:"
+                print c
+                raise ValueError
+
+    assert g.to_state_vector() == c
diff --git a/tests/test_qi_vs_qtip.py b/tests/test_qi_vs_qtip.py
new file mode 100644
index 0000000..13a7610
--- /dev/null
+++ b/tests/test_qi_vs_qtip.py
@@ -0,0 +1,9 @@
+from abp import qi
+from qutip import *
+import numpy as np
+
+def test_1():
+    q = QubitCircuit(4)
+    q.add_gate("CPHASE", 1, 2, arg_value=np.pi)
+    #print gate_sequence_product(q.propagators())
+