enrichment_test.EnrichmentTest flaky fix (#36407)

stankiewicz · web-flow · commit 7542aeabbcfb · 2025-10-06T10:32:50.000-04:00
* enrichment returns row fields in different order. simple split string makes comparison of rows flaky.

* format
diff --git a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
@@ -134,7 +134,8 @@ def test_enrichment_with_vertex_ai(self, mock_stdout):
     expected = sorted(validate_enrichment_with_vertex_ai())
 
     for i in range(len(expected)):
-      self.assertEqual(set(output[i].split(',')), set(expected[i].split(',')))
+      self.assertEqual(
+          set(output[i][4:-1].split(',')), set(expected[i][4:-1].split(',')))
 
   def test_enrichment_with_vertex_ai_legacy(self, mock_stdout):
     enrichment_with_vertex_ai_legacy()
