Add the pytest function for MPS in cuquantum.

src/qibotn/QiboCircuitToMPS.py

@@ -2,14 +2,15 @@ import cupy as cp
 import numpy as np
 
 from cuquantum import cutensornet as cutn
-from QiboCircuitConvertor import QiboCircuitToEinsum
-from MPSUtils import get_initial_mps, apply_gate
+from qibotn.QiboCircuitConvertor import QiboCircuitToEinsum
+from qibotn.MPSUtils import get_initial_mps, apply_gate
+
 
 class QiboCircuitToMPS:
-    def __init__(self,circ_qibo, gate_algo, dtype = 'complex128',rand_seed=0,):
+    def __init__(self, circ_qibo, gate_algo, dtype='complex128', rand_seed=0,):
         np.random.seed(rand_seed)
         cp.random.seed(rand_seed)
-        
+
         self.num_qubits = circ_qibo.nqubits
         self.handle = cutn.create()
         self.options = {'handle': self.handle}
@@ -20,7 +21,8 @@ class QiboCircuitToMPS:
         for (gate, qubits) in circuitconvertor.gate_tensors:
             # mapping from qubits to qubit indices
             # apply the gate in-place
-            apply_gate(self.mps_tensors, gate, qubits, algorithm=gate_algo, options=self.options)
+            apply_gate(self.mps_tensors, gate, qubits,
+                       algorithm=gate_algo, options=self.options)
 
     def __del__(self):
         cutn.destroy(self.handle)

src/qibotn/cutn.py

@@ -1,33 +1,34 @@
 from qibotn.QiboCircuitConvertor import QiboCircuitToEinsum
 from cuquantum import contract
-from MpsHelper import MPSHelper
-import cuquantum 
 from cuquantum import cutensornet as cutn
 import cupy as cp
 import numpy as np
 from qibo.models import QFT
-from QiboCircuitToMPS import QiboCircuitToMPS
-from MPSContractionHelper import MPSContractionHelper
+from qibotn.QiboCircuitToMPS import QiboCircuitToMPS
+from qibotn.MPSContractionHelper import MPSContractionHelper
 
 
 def eval(qibo_circ, datatype):
     myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
     return contract(*myconvertor.state_vector_operands())
 
+
 def eval_mps(qibo_circ, gate_algo, datatype):
-    myconvertor = QiboCircuitToMPS(qibo_circ, gate_algo, datatype)  
+    myconvertor = QiboCircuitToMPS(qibo_circ, gate_algo, dtype=datatype)
     mps_helper = MPSContractionHelper(myconvertor.num_qubits)
-    sv_mps = mps_helper.contract_state_vector(myconvertor.mps_tensors,myconvertor.options)
+    sv_mps = mps_helper.contract_state_vector(
+        myconvertor.mps_tensors, myconvertor.options)
     return sv_mps
 
+
 if __name__ == "__main__":
-    num_qubits  = 25
+    num_qubits = 25
     swaps = True
     circ_qibo = QFT(num_qubits, swaps)
-    
-    exact_gate_algorithm = {'qr_method': False, 
-                            'svd_method':{'partition': 'UV', 'abs_cutoff':1e-12}}
-    dtype="complex128"
+
+    exact_gate_algorithm = {'qr_method': False,
+                            'svd_method': {'partition': 'UV', 'abs_cutoff': 1e-12}}
+    dtype = "complex128"
     sv_mps = eval_mps(circ_qibo, exact_gate_algorithm, dtype)
     sv_reference = eval(circ_qibo, dtype)
     state_vec = np.array(circ_qibo())