Skip to content

Commit c6c815d

Browse files
authored
added the rate test for GenerateSequence (#35108)
* added the rate test for GenerateSequence * keep the master yaml
1 parent 7afeb30 commit c6c815d

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

sdks/python/apache_beam/transforms/external_transform_provider_it_test.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,18 @@ def test_run_generate_sequence(self):
122122

123123
assert_that(numbers, equal_to([i for i in range(10)]))
124124

125+
def test_run_generate_sequence_with_rate(self):
126+
provider = ExternalTransformProvider(
127+
BeamJarExpansionService(":sdks:java:io:expansion-service:shadowJar"))
128+
129+
with beam.Pipeline() as p:
130+
numbers = p | provider.GenerateSequence(
131+
start=0, end=3, rate={
132+
'elements': 1, 'seconds': 1
133+
}) | beam.Map(lambda row: row.value)
134+
135+
assert_that(numbers, equal_to([0, 1, 2]))
136+
125137

126138
@pytest.mark.xlang_wrapper_generation
127139
@unittest.skipUnless(

0 commit comments

Comments
 (0)