From 67fb52a30380c37b1907edde1cfc11b41cd86e6a Mon Sep 17 00:00:00 2001 From: Luke Baumann Date: Fri, 16 Jan 2026 11:52:59 -0800 Subject: [PATCH] Reorder imports PiperOrigin-RevId: 857248362 --- pathwaysutils/test/debug/timing_test.py | 7 +++---- pathwaysutils/test/debug/watchdog_test.py | 8 +++----- pathwaysutils/test/initialize_test.py | 7 +++---- pathwaysutils/test/lru_cache_test.py | 6 +++--- pathwaysutils/test/persistence_test.py | 7 +++---- pathwaysutils/test/plugin_executable_test.py | 6 +++--- pathwaysutils/test/profiling_test.py | 7 +++---- pathwaysutils/test/proxy_backend_test.py | 9 ++++----- 8 files changed, 25 insertions(+), 32 deletions(-) diff --git a/pathwaysutils/test/debug/timing_test.py b/pathwaysutils/test/debug/timing_test.py index 13c22cc..84b0d8d 100644 --- a/pathwaysutils/test/debug/timing_test.py +++ b/pathwaysutils/test/debug/timing_test.py @@ -17,11 +17,10 @@ import time from unittest import mock +from absl.testing import absltest +from absl.testing import parameterized from pathwaysutils.debug import timing -from google3.testing.pybase import googletest -from google3.testing.pybase import parameterized - class TimingTest(parameterized.TestCase): @@ -83,4 +82,4 @@ def my_function(): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/debug/watchdog_test.py b/pathwaysutils/test/debug/watchdog_test.py index 5ea53e9..1972304 100644 --- a/pathwaysutils/test/debug/watchdog_test.py +++ b/pathwaysutils/test/debug/watchdog_test.py @@ -19,12 +19,10 @@ import traceback from unittest import mock +from absl.testing import absltest +from absl.testing import parameterized from pathwaysutils.debug import watchdog -from google3.testing.pybase import googletest -from google3.testing.pybase import parameterized - - class WatchdogTest(parameterized.TestCase): def test_watchdog_start_join(self): with ( @@ -93,4 +91,4 @@ def test_log_thread_strack_succes(self, thread_ident, expected_log_output): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/initialize_test.py b/pathwaysutils/test/initialize_test.py index 5fc83fd..5dc90f2 100644 --- a/pathwaysutils/test/initialize_test.py +++ b/pathwaysutils/test/initialize_test.py @@ -14,12 +14,11 @@ import os +from absl.testing import absltest +from absl.testing import parameterized import jax from pathwaysutils import _initialize -from google3.testing.pybase import googletest -from google3.testing.pybase import parameterized - class InitializeTest(parameterized.TestCase): @@ -89,4 +88,4 @@ def test_persistence_enabled(self): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/lru_cache_test.py b/pathwaysutils/test/lru_cache_test.py index 82fcfcb..2acca58 100644 --- a/pathwaysutils/test/lru_cache_test.py +++ b/pathwaysutils/test/lru_cache_test.py @@ -12,12 +12,12 @@ # See the License for the specific language governing permissions and # limitations under the License. +from absl.testing import absltest import jax.extend from pathwaysutils import lru_cache -from google3.testing.pybase import googletest -class LruCacheTest(googletest.TestCase): +class LruCacheTest(absltest.TestCase): def test_cache_hits(self): x = [100] @@ -82,4 +82,4 @@ def f(i): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/persistence_test.py b/pathwaysutils/test/persistence_test.py index 0e15aef..628cc48 100644 --- a/pathwaysutils/test/persistence_test.py +++ b/pathwaysutils/test/persistence_test.py @@ -2,14 +2,13 @@ import datetime +from absl.testing import absltest import jax import numpy as np from pathwaysutils.persistence import helper -from google3.testing.pybase import googletest - -class PersistenceTest(googletest.TestCase): +class PersistenceTest(absltest.TestCase): location = "/path/to/location" name = "name" dtype = np.dtype(np.int32) @@ -106,4 +105,4 @@ def test_get_bulk_write_request(self): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/plugin_executable_test.py b/pathwaysutils/test/plugin_executable_test.py index 2b6b14e..9db8a4b 100644 --- a/pathwaysutils/test/plugin_executable_test.py +++ b/pathwaysutils/test/plugin_executable_test.py @@ -3,15 +3,15 @@ These should not exercise a specific feature that uses side channel, but rather the general logic of the class. """ +from absl.testing import absltest import jax from pathwaysutils import plugin_executable -from google3.testing.pybase import googletest PluginExecutable = plugin_executable.PluginExecutable XlaRuntimeError = jax.errors.JaxRuntimeError -class PluginExecutableTest(googletest.TestCase): +class PluginExecutableTest(absltest.TestCase): def setUp(self): jax.config.update("jax_platforms", "cpu") @@ -26,4 +26,4 @@ def test_bad_program(self): PluginExecutable("this is not json") if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/profiling_test.py b/pathwaysutils/test/profiling_test.py index ce8acd4..c61b991 100644 --- a/pathwaysutils/test/profiling_test.py +++ b/pathwaysutils/test/profiling_test.py @@ -16,13 +16,12 @@ import logging from unittest import mock +from absl.testing import absltest +from absl.testing import parameterized import jax from pathwaysutils import profiling import requests -from google3.testing.pybase import googletest -from google3.testing.pybase import parameterized - class ProfilingTest(parameterized.TestCase): """Tests for Pathways on Cloud profiling.""" @@ -457,4 +456,4 @@ def test_jax_profiler_trace_calls_patched_functions(self): if __name__ == "__main__": - googletest.main() + absltest.main() diff --git a/pathwaysutils/test/proxy_backend_test.py b/pathwaysutils/test/proxy_backend_test.py index 63502ec..11d301a 100644 --- a/pathwaysutils/test/proxy_backend_test.py +++ b/pathwaysutils/test/proxy_backend_test.py @@ -15,15 +15,14 @@ from unittest import mock +from absl.testing import absltest import jax from jax.extend import backend from pathwaysutils import jax as pw_jax from pathwaysutils import proxy_backend -from google3.testing.pybase import googletest - -class ProxyBackendTest(googletest.TestCase): +class ProxyBackendTest(absltest.TestCase): def setUp(self): super().setUp() @@ -31,7 +30,7 @@ def setUp(self): jax.config.update("jax_backend_target", "grpc://localhost:12345") backend.clear_backends() - @googletest.skip("b/408025233") + @absltest.skip("b/408025233") def test_no_proxy_backend_registration_raises_error(self): self.assertRaises(RuntimeError, backend.backends) @@ -48,4 +47,4 @@ def test_proxy_backend_registration(self): if __name__ == "__main__": - googletest.main() + absltest.main()