test: Add basic assertion on return value

qiboteam · Aug 5, 2024 · 4cb0f34 · 4cb0f34
1 parent 23e61cf
commit 4cb0f34
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/src/qibolab/backends.py b/src/qibolab/backends.py
@@ -1,4 +1,3 @@
-import traceback
 from collections import deque
 
 import numpy as np
@@ -113,10 +112,6 @@ def execute_circuit(self, circuit, initial_state=None, nshots=1000):
 
         result = MeasurementOutcomes(circuit.measurements, self, nshots=nshots)
         self.assign_measurements(measurement_map, readout)
-        samples = result.samples()
-        print(samples)
-        print(samples.dtype)
-        traceback.print_stack()
         return result
 
     def execute_circuits(self, circuits, initial_states=None, nshots=1000):

diff --git a/tests/test_backends.py b/tests/test_backends.py
@@ -33,7 +33,8 @@ def test_execute_circuit_initial_state():
 
     initial_circuit = Circuit(1)
     circuit.add(gates.GPI2(0, phi=np.pi / 2))
-    backend.execute_circuit(circuit, initial_state=initial_circuit)
+    result = backend.execute_circuit(circuit, initial_state=initial_circuit)
+    assert result.samples().size > 0
 
 
 @pytest.mark.parametrize(
@@ -50,6 +51,7 @@ def test_execute_circuit(gate, kwargs):
     nqubits = backend.platform.nqubits
     circuit = generate_circuit_with_gate(nqubits, gate, **kwargs)
     result = backend.execute_circuit(circuit, nshots=100)
+    assert result.samples().size > 0
 
 
 def test_measurement_samples():