diff --git a/sdks/python/apache_beam/io/gcp/bigquery_test.py b/sdks/python/apache_beam/io/gcp/bigquery_test.py index c263b636b57a..435fe67d02fc 100644 --- a/sdks/python/apache_beam/io/gcp/bigquery_test.py +++ b/sdks/python/apache_beam/io/gcp/bigquery_test.py @@ -2278,10 +2278,13 @@ def store_callback(table, **kwargs): num_streaming_keys=500)) with open(file_name_1) as f1, open(file_name_2) as f2: - out1 = json.load(f1) - self.assertEqual(out1['colA_values'], ['value1', 'value3']) - out2 = json.load(f2) - self.assertEqual(out2['colA_values'], ['value5']) + out1 = json.load(f1)['colA_values'] + out2 = json.load(f2)['colA_values'] + out_all = out1 + out2 + out_all.sort() + self.assertEqual(out_all, ['value1', 'value3', 'value5']) + self.assertEqual(len(out1), 2) + self.assertEqual(len(out2), 1) @unittest.skipIf(HttpError is None, 'GCP dependencies are not installed')