utils => mcm_utils.

PennyLaneAI · Jun 3, 2024 · 646a626 · 646a626
1 parent d2c9459
commit 646a626
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 17 deletions.
diff --git a/tests/devices/default_qubit/test_default_qubit_native_mcm.py b/tests/devices/default_qubit/test_default_qubit_native_mcm.py
@@ -14,9 +14,9 @@
 """Tests for default qubit preprocessing."""
 from typing import Sequence
 
+import mcm_utils
 import numpy as np
 import pytest
-import utils
 
 import pennylane as qml
 from pennylane.devices.qubit.apply_operation import MidMeasureMP, apply_mid_measure
@@ -123,7 +123,7 @@ def func(x, y):
     results1 = func1(*params)
     results2 = func2(*params)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 # pylint: disable=unused-argument
@@ -171,7 +171,7 @@ def func(x, y, z):
     results1 = func1(*params)
     results2 = func2(*params)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("postselect", [None, 0, 1])
@@ -195,7 +195,7 @@ def func(x, y, z):
     results2 = func2(*params)
 
     for measure_f, res1, res2 in zip([qml.counts, qml.expval], results1, results2):
-        utils.validate_measurements(measure_f, 5000, res1, res2)
+        mcm_utils.validate_measurements(measure_f, 5000, res1, res2)
 
 
 @pytest.mark.parametrize("shots", [None, 3000, [3000, 3001]])
@@ -226,7 +226,7 @@ def func(x, y):
     results1 = func1(*params)
     results2 = func2(*params)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("postselect", [None, 0, 1])
@@ -255,11 +255,11 @@ def func(x, y, z):
     if isinstance(shots, Sequence):
         for s, r1, r2 in zip(shots, results1, results2):
             for _r1, _r2 in zip(r1, r2):
-                utils.validate_measurements(measure_f, s, _r1, _r2)
+                mcm_utils.validate_measurements(measure_f, s, _r1, _r2)
         return
 
     for r1, r2 in zip(results1, results2):
-        utils.validate_measurements(measure_f, shots, r1, r2)
+        mcm_utils.validate_measurements(measure_f, shots, r1, r2)
 
 
 @pytest.mark.parametrize(
@@ -299,7 +299,7 @@ def func(x):
     results1 = func1(param)
     results2 = func2(param)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("shots", [None, 5000, [5000, 5001]])
@@ -341,7 +341,7 @@ def func(x):
     results1 = func1(param)
     results2 = func2(param)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("shots", [None, 5000, [5000, 5001]])
@@ -372,7 +372,7 @@ def func(x, y, z):
     results1 = func1(*params)
     results2 = func2(*params)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("shots", [7500, [5000, 5001]])
@@ -403,7 +403,7 @@ def func(x):
     results1 = func1(param)
     results2 = func2(param)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
 
 @pytest.mark.parametrize("shots", [5000])
@@ -435,7 +435,7 @@ def func(x, y):
     results1 = func1(*param)
     results2 = func2(*param)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
 
     grad1 = qml.grad(func)(*param)
     grad2 = qml.grad(func2)(*param)
@@ -467,7 +467,7 @@ def func(x, y):
     results1 = func1(*param)
     results2 = func2(*param)
 
-    utils.validate_measurements(measure_fn, shots, results1, results2, batch_size=2)
+    mcm_utils.validate_measurements(measure_fn, shots, results1, results2, batch_size=2)
 
     if measure_fn is qml.sample and postselect is None:
         for i in range(2):  # batch_size
@@ -524,7 +524,7 @@ def func(x, y):
     results1 = func1(*param)
     results2 = func2(*param)
 
-    utils.validate_measurements(qml.sample, shots, results1, results2, batch_size=None)
+    mcm_utils.validate_measurements(qml.sample, shots, results1, results2, batch_size=None)
 
     evals = obs.eigvals()
     for eig in evals:
@@ -688,4 +688,4 @@ def func(x):
     results1 = func1(param)
     results2 = func2(param)
 
-    utils.validate_measurements(measure_f, shots, results1, results2)
+    mcm_utils.validate_measurements(measure_f, shots, results1, results2)
diff --git a/tests/helpers/utils.py → tests/helpers/mcm_utils.py b/tests/helpers/utils.py → tests/helpers/mcm_utils.py
diff --git a/tests/test_compiler.py b/tests/test_compiler.py
@@ -17,9 +17,9 @@
 # pylint: disable=import-outside-toplevel
 from unittest.mock import patch
 
+import mcm_utils
 import numpy as np
 import pytest
-import utils
 
 import pennylane as qml
 from pennylane import numpy as np
@@ -810,4 +810,4 @@ def ansatz():
         if measure_f == qml.sample:
             results0 = results0[results0 != fill_in_value]
             results1 = results1[results1 != fill_in_value]
-        utils.validate_measurements(measure_f, shots, results1, results0)
+        mcm_utils.validate_measurements(measure_f, shots, results1, results0)