Avoid exceptions in concurrent tests

PiperOrigin-RevId: 834622310
Change-Id: Ibadc7f6dd1a8e135a7764a6d3667352b22fa24ec

Author: laramiel

python/tensorstore/tests/chunk_layout_test.py

@@ -176,7 +176,7 @@ def _run_threads(
 
 def test_chunk_layout_concurrent() -> None:
   """Tests concurrent access to ChunkLayout properties."""
-  layout = ts.ChunkLayout(rank=2)
+  layout = ts.ChunkLayout(chunk_shape=[10, 20], rank=2)
   stop = threading.Event()
 
   def read_props() -> None:
@@ -195,17 +195,17 @@ def update_props() -> None:
     i = 0
     while not stop.is_set():
       if (i % 2) == 0:
-        layout.update(inner_order=[0, 1], chunk_shape=[10, 20])
+        layout.update(chunk_shape=[10, 20])
       else:
-        layout.update(inner_order=[1, 0], chunk_shape=[None, None])
+        layout.update(rank=2)
       i += 1
 
   _run_threads(stop, read_props, update_props)
 
 
 def test_grid_concurrent() -> None:
   """Tests concurrent access to ChunkLayout.Grid properties."""
-  grid = ts.ChunkLayout.Grid(rank=2)
+  grid = ts.ChunkLayout.Grid(shape=[10, 10])
   stop = threading.Event()
 
   def read_props() -> None:
@@ -223,9 +223,9 @@ def update_props() -> None:
     i = 0
     while not stop.is_set():
       if (i % 2) == 0:
-        grid.update(shape=[1, 2], elements=10)
+        grid.update(shape=[10, 10])
       else:
-        grid.update(shape=[None, None], elements=None)
+        grid.update(rank=2, elements=100)
       i += 1
 
   _run_threads(stop, read_props, update_props)

python/tensorstore/tests/spec_test.py

@@ -164,7 +164,7 @@ def test_spec_concurrent_update_and_read() -> None:
   s = ts.Spec({
       "driver": "zarr",
       "kvstore": {"driver": "memory"},
-      "transform": {"input_rank": 2},
+      "schema": {"rank": 2, "dtype": "uint32"},
   })
 
   stop = threading.Event()
@@ -188,23 +188,23 @@ def read_props() -> None:
       _ = s.vindex[0]
       _ = s.T
       _ = s.transpose([1, 0])
-      _ = s.translate_to[1]
+      _ = s.translate_by[1]
 
   def update_props() -> None:
     i = 0
     while not stop.is_set():
       if (i % 2) == 0:
-        s.update(dtype=ts.uint32, shape=[10, 20])
+        s.update(shape=[10, 20], recheck_cached=False)
       else:
-        s.update(dtype=ts.int32, shape=[30, 40])
+        s.update(shape=[10, 20], recheck_cached=True)
       i += 1
 
   _run_threads(stop, read_props, update_props)
 
 
 def test_schema_concurrent_update_and_read() -> None:
   """Validates that concurrent updates and reads do not crash."""
-  s = ts.Schema(dtype=ts.int32, fill_value=42, rank=2)
+  s = ts.Schema(dtype=ts.int32, fill_value=42, shape=[10, 20])
 
   stop = threading.Event()
 
@@ -227,15 +227,15 @@ def read_props() -> None:
       _ = s.vindex[0]
       _ = s.T
       _ = s.transpose([1, 0])
-      _ = s.translate_to[1]
+      _ = s.translate_by[1]
 
   def update_props() -> None:
     i = 0
     while not stop.is_set():
       if (i % 2) == 0:
-        s.update(dtype=ts.uint32, fill_value=10)
+        s.update(fill_value=42)
       else:
-        s.update(dtype=ts.int32, fill_value=20)
+        s.update(shape=[10, 20])
       i += 1
 
   _run_threads(stop, read_props, update_props)