added number of shots for fullqst

Author: NicoRenaud

tests/test_tomography.py

@@ -38,18 +38,18 @@ def setUp(self):
 
     def test_full_qst(self):
         backend = Aer.get_backend("statevector_simulator")
-        full_qst = FullQST(self.ansatz, backend)
-        sol = full_qst.get_relative_amplitude_sign(self.parameters)
+        full_qst = FullQST(self.ansatz, backend, shots=10000)
+        _ = full_qst.get_relative_amplitude_sign(self.parameters)
         # assert np.allclose(self.ref, sol) or np.allclose(self.ref, -sol)
 
     def test_htree_qst(self):
         sampler = Sampler()
         htree_qst = HTreeQST(self.ansatz, sampler)
         sol = htree_qst.get_relative_amplitude_sign(self.parameters)
-        # assert np.allclose(self.ref, sol) or np.allclose(self.ref, -sol)
+        assert np.allclose(self.ref, sol) or np.allclose(self.ref, -sol)
 
     def test_shadow_qst(self):
         sampler = Sampler()
         shadow_qst = ShadowQST(self.ansatz, sampler, 10000)
         sol = shadow_qst.get_relative_amplitude_sign(self.parameters)
-        # assert np.allclose(self.ref, sol) or np.allclose(self.ref, -sol)
+        assert np.allclose(self.ref, sol) or np.allclose(self.ref, -sol)

vqls_prototype/tomography/qst.py

@@ -3,9 +3,10 @@
 
 
 class FullQST:
-    def __init__(self, circuit, backend):
+    def __init__(self, circuit, backend, shots=1000):
         self.backend = backend
         self.circuit = circuit
+        self.shots = shots
 
     def get_relative_amplitude_sign(self, parameters):
         """_summary_
@@ -50,7 +51,7 @@ def get_density_matrix(self, parameters):
             _type_: _description_
         """
         qstexp1 = StateTomography(self.circuit.bind_parameters(parameters))
-        qstdata1 = qstexp1.run(self.backend).block_for_results()
+        qstdata1 = qstexp1.run(self.backend, shots=self.shots).block_for_results()
         return qstdata1.analysis_results("state").value.data.real
 
     def get_statevector(self, parameters):