From 61c36618017b03b979877bbae6a6eb42faf38df7 Mon Sep 17 00:00:00 2001 From: Vikas Kedigehalli Date: Mon, 19 Jun 2017 11:24:14 -0700 Subject: [PATCH] Fix dataflow runner test to call pipeline.run instead of runner.run --- .../apache_beam/runners/dataflow/dataflow_runner_test.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/sdks/python/apache_beam/runners/dataflow/dataflow_runner_test.py b/sdks/python/apache_beam/runners/dataflow/dataflow_runner_test.py index 6cc5814a5fbc..a9b8fdb2a24c 100644 --- a/sdks/python/apache_beam/runners/dataflow/dataflow_runner_test.py +++ b/sdks/python/apache_beam/runners/dataflow/dataflow_runner_test.py @@ -116,10 +116,7 @@ def test_streaming_create_translation(self): self.default_properties.append("--streaming") p = Pipeline(remote_runner, PipelineOptions(self.default_properties)) p | ptransform.Create([1]) # pylint: disable=expression-not-assigned - remote_runner.job = apiclient.Job(p._options) - # Performing configured PTransform overrides here. - p.replace_all(DataflowRunner._PTRANSFORM_OVERRIDES) - super(DataflowRunner, remote_runner).run(p) + p.run() job_dict = json.loads(str(remote_runner.job)) self.assertEqual(len(job_dict[u'steps']), 2)