Add __all__ tags to modules in package apache_beam/testing
Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/0f910b43 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/0f910b43 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/0f910b43 Branch: refs/heads/release-2.0.0 Commit: 0f910b43cbccca73d3b492560b9fc0d3f0901e21 Parents: a6f5888 Author: Charles Chen <[email protected]> Authored: Wed May 10 23:20:20 2017 -0700 Committer: Ahmet Altay <[email protected]> Committed: Thu May 11 16:54:12 2017 -0700 ---------------------------------------------------------------------- sdks/python/apache_beam/testing/pipeline_verifiers.py | 8 ++++++++ sdks/python/apache_beam/testing/test_pipeline.py | 5 +++++ sdks/python/apache_beam/testing/test_stream.py | 14 +++++++++++++- sdks/python/apache_beam/testing/test_utils.py | 6 +++++- 4 files changed, 31 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/0f910b43/sdks/python/apache_beam/testing/pipeline_verifiers.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/testing/pipeline_verifiers.py b/sdks/python/apache_beam/testing/pipeline_verifiers.py index 5a6082a..a08eb54 100644 --- a/sdks/python/apache_beam/testing/pipeline_verifiers.py +++ b/sdks/python/apache_beam/testing/pipeline_verifiers.py @@ -32,6 +32,14 @@ from apache_beam.runners.runner import PipelineState from apache_beam.testing import test_utils as utils from apache_beam.utils import retry + +__all__ = [ + 'PipelineStateMatcher', + 'FileChecksumMatcher', + 'retry_on_io_error_and_server_error', + ] + + try: from apitools.base.py.exceptions import HttpError except ImportError: http://git-wip-us.apache.org/repos/asf/beam/blob/0f910b43/sdks/python/apache_beam/testing/test_pipeline.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/testing/test_pipeline.py b/sdks/python/apache_beam/testing/test_pipeline.py index 20f4839..13b1639 100644 --- a/sdks/python/apache_beam/testing/test_pipeline.py +++ b/sdks/python/apache_beam/testing/test_pipeline.py @@ -27,6 +27,11 @@ from apache_beam.options.pipeline_options import PipelineOptions from nose.plugins.skip import SkipTest +__all__ = [ + 'TestPipeline', + ] + + class TestPipeline(Pipeline): """TestPipeline class is used inside of Beam tests that can be configured to run against pipeline runner. http://git-wip-us.apache.org/repos/asf/beam/blob/0f910b43/sdks/python/apache_beam/testing/test_stream.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/testing/test_stream.py b/sdks/python/apache_beam/testing/test_stream.py index 7ae27b7..a06bcd0 100644 --- a/sdks/python/apache_beam/testing/test_stream.py +++ b/sdks/python/apache_beam/testing/test_stream.py @@ -15,7 +15,10 @@ # limitations under the License. # -"""Provides TestStream for verifying streaming runner semantics.""" +"""Provides TestStream for verifying streaming runner semantics. + +For internal use only; no backwards-compatibility guarantees. +""" from abc import ABCMeta from abc import abstractmethod @@ -28,6 +31,15 @@ from apache_beam.utils import timestamp from apache_beam.utils.windowed_value import WindowedValue +__all__ = [ + 'Event', + 'ElementEvent', + 'WatermarkEvent', + 'ProcessingTimeEvent', + 'TestStream', + ] + + class Event(object): """Test stream event to be emitted during execution of a TestStream.""" http://git-wip-us.apache.org/repos/asf/beam/blob/0f910b43/sdks/python/apache_beam/testing/test_utils.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/testing/test_utils.py b/sdks/python/apache_beam/testing/test_utils.py index 666207e..9feb80e 100644 --- a/sdks/python/apache_beam/testing/test_utils.py +++ b/sdks/python/apache_beam/testing/test_utils.py @@ -15,7 +15,10 @@ # limitations under the License. # -"""Utility methods for testing""" +"""Utility methods for testing + +For internal use only; no backwards-compatibility guarantees. +""" import hashlib import imp @@ -23,6 +26,7 @@ from mock import Mock, patch from apache_beam.utils import retry + DEFAULT_HASHING_ALG = 'sha1'
