fix: updated test

Author: ringoldsdev

tests/test_parallel_transformer.py

@@ -84,16 +84,18 @@ def test_tap_side_effects(self):
 
 
 def safe_increment(x: int, ctx: PipelineContext) -> int:
-  current_items = ctx["items"]
-  time.sleep(0.001)
-  ctx["items"] = current_items + 1
+  with ctx["lock"]:
+    current_items = ctx["items"]
+    time.sleep(0.001)
+    ctx["items"] = current_items + 1
   return x * 2
 
 
 def update_stats(x: int, ctx: PipelineContext) -> int:
-  ctx["total_sum"] += x
-  ctx["item_count"] += 1
-  ctx["max_value"] = max(ctx["max_value"], x)
+  with ctx["lock"]:
+    ctx["total_sum"] += x
+    ctx["item_count"] += 1
+    ctx["max_value"] = max(ctx["max_value"], x)
   return x * 3
 
 

tests/test_threaded_transformer.py

@@ -15,13 +15,6 @@
 class TestThreadedTransformerBasics:
   """Test core parallel transformer functionality."""
 
-  def test_initialization_defaults(self):
-    """Test parallel transformer initialization with default values."""
-    transformer = ThreadedTransformer[int, int]()
-    assert transformer.max_workers == 4
-    assert transformer.ordered is True
-    assert transformer.chunk_size == 1000
-
   def test_initialization_custom_parameters(self):
     """Test initialization with custom parameters."""
     transformer = ThreadedTransformer[int, int](max_workers=8, ordered=False, chunk_size=500)