fix #2499 (comment)

unitaryfund · Sep 13, 2024 · 003d3a1 · 003d3a1
1 parent 7708f50
commit 003d3a1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/mitiq/lre/tests/test_lre.py b/mitiq/lre/tests/test_lre.py
@@ -27,12 +27,18 @@ def execute(circuit, noise_level=0.025):
 noisy_val = execute(test_cirq)
 
 
-def test_lre_exp_value():
+@pytest.mark.parametrize(
+    "input_degree, input_fold_multiplier", [(2, 2), (2, 3), (3, 4)]
+)
+def test_lre_exp_value(input_degree, input_fold_multiplier):
     """Verify LRE executors work as expected."""
     ideal_val = execute(test_cirq, noise_level=0)
     assert abs(ideal_val - noisy_val) > 0
     lre_exp_val = execute_with_lre(
-        test_cirq, execute, degree=2, fold_multiplier=2
+        test_cirq,
+        execute,
+        degree=input_degree,
+        fold_multiplier=input_fold_multiplier,
     )
     assert lre_exp_val > noisy_val