From 48333e4a95891d5adb9c3d95f0d6f4d23f3e5067 Mon Sep 17 00:00:00 2001 From: mhucka Date: Fri, 27 Mar 2026 04:33:36 +0000 Subject: [PATCH 1/2] Add tests for `spin_system.unique_name()` This PR adds a test class `UniqueNameTest` to `tensorflow_quantum/datasets/spin_system_test.py` to test the `unique_name` generator function in `spin_system.py`. --- tensorflow_quantum/datasets/spin_system_test.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tensorflow_quantum/datasets/spin_system_test.py b/tensorflow_quantum/datasets/spin_system_test.py index cfb20ee4c..cd94c842f 100644 --- a/tensorflow_quantum/datasets/spin_system_test.py +++ b/tensorflow_quantum/datasets/spin_system_test.py @@ -328,6 +328,19 @@ def test_param_resolver(self): rtol=1e-3) +class UniqueNameTest(tf.test.TestCase): + """Testing unique_name.""" + + def test_unique_name(self): + """Test that unique_name generates the correct sequence.""" + gen = spin_system.unique_name() + self.assertEqual(next(gen), "theta_0") + self.assertEqual(next(gen), "theta_1") + self.assertEqual(next(gen), "theta_2") + for i in range(3, 100): + self.assertEqual(next(gen), "theta_" + str(i)) + + if __name__ == '__main__': tf.test.main() From 30504219f0aef147f93f4996301e01d3b17e01e3 Mon Sep 17 00:00:00 2001 From: Michael Hucka Date: Thu, 26 Mar 2026 21:44:23 -0700 Subject: [PATCH 2/2] Update tensorflow_quantum/datasets/spin_system_test.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- tensorflow_quantum/datasets/spin_system_test.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tensorflow_quantum/datasets/spin_system_test.py b/tensorflow_quantum/datasets/spin_system_test.py index cd94c842f..682870c25 100644 --- a/tensorflow_quantum/datasets/spin_system_test.py +++ b/tensorflow_quantum/datasets/spin_system_test.py @@ -334,11 +334,8 @@ class UniqueNameTest(tf.test.TestCase): def test_unique_name(self): """Test that unique_name generates the correct sequence.""" gen = spin_system.unique_name() - self.assertEqual(next(gen), "theta_0") - self.assertEqual(next(gen), "theta_1") - self.assertEqual(next(gen), "theta_2") - for i in range(3, 100): - self.assertEqual(next(gen), "theta_" + str(i)) + for i in range(100): + self.assertEqual(next(gen), f"theta_{i}") if __name__ == '__main__':