Got the repro

apache · Sep 26, 2024 · 65e3825 · 65e3825
1 parent 3a472e0
commit 65e3825
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py b/sdks/python/apache_beam/io/gcp/bigquery_write_it_test.py
@@ -470,7 +470,7 @@ def test_big_query_write_insert_non_transient_api_call_error(self):
     input_data = [{
         'number': 1,
         'str': 'some_string',
-    }]*500
+    }]
 
     table_schema = {
         "fields": [{
@@ -483,14 +483,15 @@ def test_big_query_write_insert_non_transient_api_call_error(self):
     bq_result_errors = [({
         'number': 1,
         'str': 'some_string',
-    }, "Not Found")]*500
+    }, "Not Found")]
 
     args = self.test_pipeline.get_full_options_as_args()
 
     with beam.Pipeline(argv=args) as p:
       # pylint: disable=expression-not-assigned
       errors = (
           p | 'create' >> beam.Create(input_data)
+          | beam.WindowInto(beam.transforms.window.FixedWindows(10))
           | 'write' >> beam.io.WriteToBigQuery(
               table_id,
               schema=table_schema,