Fix pytest

qiboteam · Mar 25, 2022 · 6196f49 · 6196f49
1 parent 44743b0
commit 6196f49
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/qibo/tests/test_core_circuit_noise.py b/src/qibo/tests/test_core_circuit_noise.py
@@ -189,4 +189,6 @@ def test_circuit_add_sampling(backend):
     for _ in range(10):
         target_samples.append(circ(nshots=1).samples())
 
-    K.assert_allclose(samples[:, 0], target_samples[:, 0, 0])
+    target_samples = np.stack(target_samples)
+
+    K.assert_allclose(samples, target_samples[:, 0])