Remove nondeterminism from expansion tests (#33082)

apache · Nov 11, 2024 · 97ae5aa · 97ae5aa
1 parent fca0bea
commit 97ae5aa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/transforms/external_test.py b/sdks/python/apache_beam/transforms/external_test.py
@@ -52,6 +52,7 @@
 from apache_beam.typehints.native_type_compatibility import convert_to_beam_type
 from apache_beam.utils import proto_utils
 from apache_beam.utils.subprocess_server import JavaJarServer
+from apache_beam.utils.subprocess_server import SubprocessServer
 
 # Protect against environments where apitools library is not available.
 # pylint: disable=wrong-import-order, wrong-import-position
@@ -718,6 +719,9 @@ def test_implicit_builder_with_constructor_method(self):
 
 
 class JavaJarExpansionServiceTest(unittest.TestCase):
+  def setUp(self):
+    SubprocessServer._cache._live_owners = set()
+
   def test_classpath(self):
     with tempfile.TemporaryDirectory() as temp_dir:
       try: