feat: test for observable default value

qiboteam · Jan 28, 2025 · 6e748c9 · 6e748c9
1 parent 8f49487
commit 6e748c9
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 63 deletions.
diff --git a/tests/#test_models_decoding.py# b/tests/#test_models_decoding.py#
diff --git a/tests/.#test_models_decoding.py b/tests/.#test_models_decoding.py
diff --git a/tests/test_models_decoding.py b/tests/test_models_decoding.py
@@ -28,24 +28,32 @@ def test_state_layer(backend):
 
 
 @pytest.mark.parametrize("nshots", [None, 10000])
-def test_expectation_layer(backend, nshots):
+@pytest.mark.parametrize(
+    "observable",
+    [
+        None,
+        lambda n, b: hamiltonians.SymbolicHamiltonian(
+            sum([Z(i) for i in range(n)]), nqubits=n, backend=b
+        ),
+    ],
+)
+def test_expectation_layer(backend, nshots, observable):
     backend.set_seed(42)
     rng = np.random.default_rng(42)
     nqubits = 5
 
     c = random_clifford(nqubits, seed=rng, backend=backend)
-    observable = hamiltonians.SymbolicHamiltonian(
-        sum([Z(i) for i in range(nqubits)]),
-        nqubits=nqubits,
-        backend=backend,
-    )
+    if observable is not None:
+        observable = observable(nqubits, backend)
     layer = dec.Expectation(
         nqubits,
         observable=observable,
         nshots=nshots,
         backend=backend,
     )
     layer_expv = layer(c)
+    if observable is None:
+        observable = hamiltonians.Z(nqubits, dense=False, backend=backend)
     expv = (
         observable.expectation(backend.execute_circuit(c).state())
         if nshots is None