more tests

Gate_test.py

@@ -1,6 +1,6 @@
 from QElephant.QuBit import *
 
-## for IQuBit
+## MuBit
 mq = MuBit(2)
 q = mq[1]
 
@@ -10,3 +10,21 @@ H(q)
 assert mq._MuBit__state == [1/math.sqrt(2), -1/math.sqrt(2), 0, 0]
 CX(mq, 0, 1)
 assert mq._MuBit__state == [1/math.sqrt(2), -1/math.sqrt(2), 0, 0]
+
+## QuBit
+q = QuBit()
+
+X(q)
+assert q._QuBit__state == [0, 1]
+H(q)
+assert q._QuBit__state == [1/math.sqrt(2), -1/math.sqrt(2)]
+SQRTX(q)
+assert q._QuBit__state == [complex(0, 1/math.sqrt(2)), -complex(0, 1/math.sqrt(2))]
+Y(q)
+assert q._QuBit__state == [-1/math.sqrt(2), -1/math.sqrt(2)]
+Z(q)
+assert q._QuBit__state == [-1/math.sqrt(2), 1/math.sqrt(2)]
+S(q)
+assert q._QuBit__state == [-1/math.sqrt(2), complex(0, 1/math.sqrt(2))]
+T(q)
+assert q._QuBit__state == [-1/math.sqrt(2), complex(0, 1/math.sqrt(2))*math.e**(complex(0, math.pi/4))]