Test for non-global window

hjtran · Jan 8, 2024 · c96c218 · c96c218
1 parent 04c3b2d
commit c96c218
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/transforms/util_test.py b/sdks/python/apache_beam/transforms/util_test.py
@@ -784,6 +784,17 @@ def to_type_hint(self):
           | beam.Map(lambda foobar: foobar.attr))
       assert_that(result, equal_to(['zab']))
 
+    # Test a streaming pipeline
+    options = PipelineOptions()
+    with beam.Pipeline(options=options) as p:
+      result = (
+          p
+          | beam.Create([FooBar('zab')], reshuffle=False)
+          | beam.WindowInto(window.SlidingWindows(1, 1))
+          | beam.Reshuffle()
+          | beam.Map(lambda foobar: foobar.attr))
+      assert_that(result, equal_to(['zab']))
+
   @pytest.mark.it_validatesrunner
   def test_reshuffle_preserves_timestamps(self):
     with TestPipeline() as pipeline: