Merge pull request #15092 from apache/revert-14819-aaltay-patch-1

aaltay · web-flow · commit 22205ee1a845 · 2021-06-28T17:16:25.000-07:00
diff --git a/sdks/python/apache_beam/runners/portability/fn_api_runner/fn_runner_test.py b/sdks/python/apache_beam/runners/portability/fn_api_runner/fn_runner_test.py
@@ -825,15 +825,11 @@ def raise_error(x):
             | beam.Create(['a', 'b'])
             | 'StageA' >> beam.Map(lambda x: x)
             | 'StageB' >> beam.Map(lambda x: x)
-            | 'FusionBreakBeforeRaise' >> beam.Reshuffle()
             | 'StageC' >> beam.Map(raise_error)
-            | 'FusionBreakAfterRaise' >> beam.Reshuffle()
             | 'StageD' >> beam.Map(lambda x: x))
     message = e_cm.exception.args[0]
     self.assertIn('StageC', message)
-    self.assertNotIn('StageA', message)
     self.assertNotIn('StageB', message)
-    self.assertNotIn('StageD', message)
 
   def test_error_traceback_includes_user_code(self):
     def first(x):
