diff --git a/sdks/python/apache_beam/transforms/external_transform_provider_it_test.py b/sdks/python/apache_beam/transforms/external_transform_provider_it_test.py index d1c5cbfa8e9a..5f115e6d44e9 100644 --- a/sdks/python/apache_beam/transforms/external_transform_provider_it_test.py +++ b/sdks/python/apache_beam/transforms/external_transform_provider_it_test.py @@ -122,6 +122,18 @@ def test_run_generate_sequence(self): assert_that(numbers, equal_to([i for i in range(10)])) + def test_run_generate_sequence_with_rate(self): + provider = ExternalTransformProvider( + BeamJarExpansionService(":sdks:java:io:expansion-service:shadowJar")) + + with beam.Pipeline() as p: + numbers = p | provider.GenerateSequence( + start=0, end=3, rate={ + 'elements': 1, 'seconds': 1 + }) | beam.Map(lambda row: row.value) + + assert_that(numbers, equal_to([0, 1, 2])) + @pytest.mark.xlang_wrapper_generation @unittest.skipUnless(