Skip to content

Add tests for channels for util.py's gate_approx_eq #1028

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
mhucka wants to merge 4 commits into
base: master
Choose a base branch
from
Open

Add tests for channels for util.py's gate_approx_eq #1028

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
41d9b9f
Add tests for channels in gate_approx_eq
mhucka Mar 27, 2026
ae899a6
Improve code
mhucka Mar 27, 2026
e147592
Fix incorrect f-string placement
mhucka Mar 27, 2026
a1bafdd
Merge branch 'master' into add-tests-gate-approx-eq
mhucka Mar 31, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
132 changes: 130 additions & 2 deletions tensorflow_quantum/python/util_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@

def _single_to_tensor(item):
if not isinstance(item, (cirq.PauliSum, cirq.PauliString, cirq.Circuit)):
raise TypeError(
f"Item must be a Circuit or PauliSum. Got {type(item)}.")
raise TypeError("Item must be a Circuit or PauliSum. Got {}.".format(
type(item)))
if isinstance(item, (cirq.PauliSum, cirq.PauliString)):
return serializer.serialize_paulisum(item).SerializeToString(
deterministic=True)
Expand Down Expand Up @@ -347,6 +347,134 @@ def test_gate_approx_eq(self):
util.gate_approx_eq(
cirq.X, cirq.ops.ControlledGate(cirq.X, 2, [1, 0], [2, 2])))

def test_gate_approx_eq_channels(self):
"""Check valid TFQ channels for approximate equality."""
atol = 1e-2

# DepolarizingChannel
self.assertTrue(
util.gate_approx_eq(cirq.DepolarizingChannel(0.1),
cirq.DepolarizingChannel(0.1),
atol=atol))
self.assertTrue(
util.gate_approx_eq(cirq.DepolarizingChannel(0.1),
cirq.DepolarizingChannel(0.105),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.DepolarizingChannel(0.1),
cirq.DepolarizingChannel(0.2),
atol=atol))

# AsymmetricDepolarizingChannel
self.assertTrue(
util.gate_approx_eq(
cirq.AsymmetricDepolarizingChannel(0.1, 0.2, 0.3),
cirq.AsymmetricDepolarizingChannel(0.1, 0.2, 0.3),
atol=atol))
self.assertTrue(
util.gate_approx_eq(
cirq.AsymmetricDepolarizingChannel(0.1, 0.2, 0.3),
cirq.AsymmetricDepolarizingChannel(0.105, 0.195, 0.305),
atol=atol))
self.assertFalse(
util.gate_approx_eq(
cirq.AsymmetricDepolarizingChannel(0.1, 0.2, 0.3),
cirq.AsymmetricDepolarizingChannel(0.2, 0.2, 0.3),
atol=atol))

# GeneralizedAmplitudeDampingChannel
self.assertTrue(
util.gate_approx_eq(
cirq.GeneralizedAmplitudeDampingChannel(0.1, 0.2),
cirq.GeneralizedAmplitudeDampingChannel(0.1, 0.2),
atol=atol))
self.assertTrue(
util.gate_approx_eq(
cirq.GeneralizedAmplitudeDampingChannel(0.1, 0.2),
cirq.GeneralizedAmplitudeDampingChannel(0.105, 0.205),
atol=atol))
self.assertFalse(
util.gate_approx_eq(
cirq.GeneralizedAmplitudeDampingChannel(0.1, 0.2),
cirq.GeneralizedAmplitudeDampingChannel(0.2, 0.2),
atol=atol))

# AmplitudeDampingChannel
self.assertTrue(
util.gate_approx_eq(cirq.AmplitudeDampingChannel(0.1),
cirq.AmplitudeDampingChannel(0.1),
atol=atol))
self.assertTrue(
util.gate_approx_eq(cirq.AmplitudeDampingChannel(0.1),
cirq.AmplitudeDampingChannel(0.105),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.AmplitudeDampingChannel(0.1),
cirq.AmplitudeDampingChannel(0.2),
atol=atol))

# ResetChannel
self.assertTrue(
util.gate_approx_eq(cirq.ResetChannel(),
cirq.ResetChannel(),
atol=atol))

# PhaseDampingChannel
self.assertTrue(
util.gate_approx_eq(cirq.PhaseDampingChannel(0.1),
cirq.PhaseDampingChannel(0.1),
atol=atol))
self.assertTrue(
util.gate_approx_eq(cirq.PhaseDampingChannel(0.1),
cirq.PhaseDampingChannel(0.105),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.PhaseDampingChannel(0.1),
cirq.PhaseDampingChannel(0.2),
atol=atol))

# PhaseFlipChannel
self.assertTrue(
util.gate_approx_eq(cirq.PhaseFlipChannel(0.1),
cirq.PhaseFlipChannel(0.1),
atol=atol))
self.assertTrue(
util.gate_approx_eq(cirq.PhaseFlipChannel(0.1),
cirq.PhaseFlipChannel(0.105),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.PhaseFlipChannel(0.1),
cirq.PhaseFlipChannel(0.2),
atol=atol))

# BitFlipChannel
self.assertTrue(
util.gate_approx_eq(cirq.BitFlipChannel(0.1),
cirq.BitFlipChannel(0.1),
atol=atol))
self.assertTrue(
util.gate_approx_eq(cirq.BitFlipChannel(0.1),
cirq.BitFlipChannel(0.105),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.BitFlipChannel(0.1),
cirq.BitFlipChannel(0.2),
atol=atol))

# Mismatched types
self.assertFalse(
util.gate_approx_eq(cirq.DepolarizingChannel(0.1),
cirq.BitFlipChannel(0.1),
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.DepolarizingChannel(0.1),
cirq.X,
atol=atol))
self.assertFalse(
util.gate_approx_eq(cirq.X,
cirq.DepolarizingChannel(0.1),
atol=atol))

def test_gate_approx_eq_error(self):
"""Confirms that bad inputs cause an error to be raised."""
# junk
Expand Down
Loading