From 3bf1f98322ba2737e950793847deb14a8ca609fc Mon Sep 17 00:00:00 2001 From: tvalentyn Date: Mon, 30 Jul 2018 19:49:46 -0700 Subject: [PATCH] Don't rely on order of elements in a PCollection after GBK in test_reshuffle_windows_unchanged. --- sdks/python/apache_beam/transforms/util_test.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/sdks/python/apache_beam/transforms/util_test.py b/sdks/python/apache_beam/transforms/util_test.py index 18efc44e4d34..d834a1c5efea 100644 --- a/sdks/python/apache_beam/transforms/util_test.py +++ b/sdks/python/apache_beam/transforms/util_test.py @@ -231,11 +231,11 @@ def test_reshuffle_timestamps_unchanged(self): def test_reshuffle_windows_unchanged(self): pipeline = TestPipeline() data = [(1, 1), (2, 1), (3, 1), (1, 2), (2, 2), (1, 4)] - expected_data = [TestWindowedValue(v, t, [w]) for (v, t, w) in - [((1, [2, 1]), 4.0, IntervalWindow(1.0, 4.0)), - ((2, [2, 1]), 4.0, IntervalWindow(1.0, 4.0)), - ((3, [1]), 3.0, IntervalWindow(1.0, 3.0)), - ((1, [4]), 6.0, IntervalWindow(4.0, 6.0))]] + expected_data = [TestWindowedValue(v, t, [w]) for (v, t, w) in [ + ((1, contains_in_any_order([2, 1])), 4.0, IntervalWindow(1.0, 4.0)), + ((2, contains_in_any_order([2, 1])), 4.0, IntervalWindow(1.0, 4.0)), + ((3, [1]), 3.0, IntervalWindow(1.0, 3.0)), + ((1, [4]), 6.0, IntervalWindow(4.0, 6.0))]] before_reshuffle = (pipeline | 'start' >> beam.Create(data) | 'add_timestamp' >> beam.Map(