From e0aa28b064f53fe46d3e36ea6f2494b12169a312 Mon Sep 17 00:00:00 2001
From: Didictateur <decosse.adrien@gmail.com>
Date: Thu, 26 Oct 2023 15:42:17 +0200
Subject: [PATCH] tests

---
 Matrix_test.py |  1 -
 QuBit_test.py  | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 49 insertions(+), 1 deletion(-)
 create mode 100644 QuBit_test.py

diff --git a/Matrix_test.py b/Matrix_test.py
index 7f44363..97095ce 100644
--- a/Matrix_test.py
+++ b/Matrix_test.py
@@ -1,5 +1,4 @@
 from QElephant.Matrix import *
-from pytest import *
 
 m = Matrix([[1, 2, 3], [4, 5, 6]])
 
diff --git a/QuBit_test.py b/QuBit_test.py
new file mode 100644
index 0000000..8f47333
--- /dev/null
+++ b/QuBit_test.py
@@ -0,0 +1,49 @@
+from QElephant.QuBit import *
+
+q = QuBit(0, 1)
+
+assert q._QuBit__state == [0, 1]
+assert q._QuBit__intricated == False
+assert q.is_intricated() == False
+assert q.get_Mubit() == None
+assert str(q) == "0 |0> + 1 |1>"
+q._QuBit__apply(Matrix([[1, 2], [3, 4]]))
+assert q._QuBit__state == [2, 4]
+q.reset()
+assert q._QuBit__state == [1, 0]
+
+mq = MuBit(2)
+
+assert mq._MuBit__state == [1, 0, 0, 0]
+assert mq._MuBit__n == 2
+assert mq.get_size() == 2
+assert str(mq) == "1 |00>\n0 |01>\n0 |10>\n0 |11>\n"
+mq._MuBit__state = [0.5]*4
+mq._MuBit__set(1, 1)
+assert mq._MuBit__state == [0, 1/math.sqrt(2), 0, 1/math.sqrt(2)]
+
+q = mq[0]
+assert q._IQuBit__n == 0
+assert q._IQuBit__muBit == mq
+assert q._IQuBit__intricated == True
+assert q.is_intricated() == True
+assert q.get_Mubit() == (mq, 0)
+assert str(q) == "0.707 |0> + 0.707 |1>"
+q._IQuBit__apply(Matrix([[1, 1], [1, -1]]))
+assert mq._MuBit__state == [0, 2/math.sqrt(2), 0, 0]
+q.reset()
+assert str(q) == "1.0 |0> + 0.0 |1>"
+
+mq._MuBit__state = [1, 1, 1, 1]
+mq.reset()
+assert mq._MuBit__state == [1, 0, 0, 0]
+mq._MuBit__apply(Matrix([[0, 1], [1, 0]]), 0)
+assert mq._MuBit__state == [0, 0, 1, 0]
+mq._MuBit__mapply(Matrix([[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 0], [0, 0, 2, 0]]), 0)
+assert mq._MuBit__state == [0, 0, 1, 2]
+mq._MuBit__state = [0.5, 1, 2, 0]
+assert mq._MuBit__getProb(0) == 1.25
+mq._MuBit__SWITCH(0)
+assert mq._MuBit__state == [0.5, 2, 1, 0]
+mq = MuBit.intricateThem(QuBit(1,0), QuBit(0,1), QuBit(1,0), QuBit(0,1))
+assert mq._MuBit__state == [0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
\ No newline at end of file