Serialize SingleQubitCliffordGate (#4421)

Closes #3980 I tried to mimic the `PauliString` Class for serialization - https://github.com/quantumlib/Cirq/blob/4aeb719c2ff9f2394a564237886fb573c830ecad/cirq-core/cirq/ops/pauli_string.py#L175-L185
quantumlib · Aug 24, 2021 · 58cda7a · 58cda7a
1 parent 932f065
commit 58cda7a
Show file tree

Hide file tree

Showing 6 changed files with 120 additions and 1 deletion.
diff --git a/cirq-core/cirq/json_resolver_cache.py b/cirq-core/cirq/json_resolver_cache.py
@@ -128,6 +128,7 @@ def two_qubit_matrix_gate(matrix):
         'QuantumFourierTransformGate': cirq.QuantumFourierTransformGate,
         'RepetitionsStoppingCriteria': cirq.work.RepetitionsStoppingCriteria,
         'ResetChannel': cirq.ResetChannel,
+        'SingleQubitCliffordGate': cirq.SingleQubitCliffordGate,
         'SingleQubitMatrixGate': single_qubit_matrix_gate,
         'SingleQubitPauliStringGateOperation': cirq.SingleQubitPauliStringGateOperation,
         'SingleQubitReadoutCalibrationResult': cirq.experiments.SingleQubitReadoutCalibrationResult,

diff --git a/cirq-core/cirq/ops/clifford_gate.py b/cirq-core/cirq/ops/clifford_gate.py
@@ -436,6 +436,21 @@ def __repr__(self) -> str:
             f'Y:{y_sign}{y.to!s}, Z:{z_sign}{z.to!s})'
         )
 
+    @classmethod
+    def _from_json_dict_(cls, _rotation_map, _inverse_map, **kwargs):
+        return cls(
+            _rotation_map=dict([[k, PauliTransform(*v)] for k, v in _rotation_map]),
+            _inverse_map=dict([[k, PauliTransform(*v)] for k, v in _inverse_map]),
+        )
+
+    def _json_dict_(self) -> Dict[str, Any]:
+        return {
+            'cirq_type': self.__class__.__name__,
+            # JSON requires mappings to have string keys.
+            '_rotation_map': list(self._rotation_map.items()),
+            '_inverse_map': list(self._inverse_map.items()),
+        }
+
     def _circuit_diagram_info_(
         self, args: 'cirq.CircuitDiagramInfoArgs'
     ) -> 'cirq.CircuitDiagramInfo':

diff --git a/cirq-core/cirq/ops/clifford_gate_test.py b/cirq-core/cirq/ops/clifford_gate_test.py
@@ -443,6 +443,11 @@ def test_commutes_single_qubit_gate(gate, other):
     assert_allclose_up_to_global_phase(mat, mat_swap, rtol=1e-7, atol=1e-7)
 
 
+@pytest.mark.parametrize('gate', _all_clifford_gates())
+def test_parses_single_qubit_gate(gate):
+    assert gate == cirq.read_json(json_text=(cirq.to_json(gate)))
+
+
 @pytest.mark.parametrize(
     'gate,pauli,half_turns',
     itertools.product(_all_clifford_gates(), _paulis, (1.0, 0.25, 0.5, -0.5)),

diff --git a/cirq-core/cirq/protocols/json_test_data/SingleQubitCliffordGate.json b/cirq-core/cirq/protocols/json_test_data/SingleQubitCliffordGate.json
@@ -0,0 +1,97 @@
+{
+  "cirq_type": "SingleQubitCliffordGate",
+  "_rotation_map": [
+    [
+      {
+        "cirq_type": "_PauliX",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliX",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        false
+      ]
+    ],
+    [
+      {
+        "cirq_type": "_PauliZ",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliY",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        false
+      ]
+    ],
+    [
+      {
+        "cirq_type": "_PauliY",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliZ",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        true
+      ]
+    ]
+  ],
+  "_inverse_map": [
+    [
+      {
+        "cirq_type": "_PauliX",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliX",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        false
+      ]
+    ],
+    [
+      {
+        "cirq_type": "_PauliY",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliZ",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        false
+      ]
+    ],
+    [
+      {
+        "cirq_type": "_PauliZ",
+        "exponent": 1.0,
+        "global_shift": 0.0
+      },
+      [
+        {
+          "cirq_type": "_PauliY",
+          "exponent": 1.0,
+          "global_shift": 0.0
+        },
+        true
+      ]
+    ]
+  ]
+}
diff --git a/cirq-core/cirq/protocols/json_test_data/SingleQubitCliffordGate.repr b/cirq-core/cirq/protocols/json_test_data/SingleQubitCliffordGate.repr
@@ -0,0 +1,2 @@
+cirq.SingleQubitCliffordGate(_rotation_map={cirq.X: cirq.PauliTransform(to=cirq.X, flip=False), cirq.Z: cirq.PauliTransform(to=cirq.Y, flip=False), cirq.Y: cirq.PauliTransform(to=cirq.Z, flip=True)},
+_inverse_map={cirq.X: cirq.PauliTransform(to=cirq.X, flip=False), cirq.Y: cirq.PauliTransform(to=cirq.Z, flip=False), cirq.Z: cirq.PauliTransform(to=cirq.Y, flip=True)})
diff --git a/cirq-core/cirq/protocols/json_test_data/spec.py b/cirq-core/cirq/protocols/json_test_data/spec.py
@@ -64,7 +64,6 @@
         'QuilFormatter',
         'QuilOutput',
         'SimulationTrialResult',
-        'SingleQubitCliffordGate',
         'SparseSimulatorStep',
         'StateVectorMixin',
         'TextDiagramDrawer',