Repository: beam Updated Branches: refs/heads/master 674ae9a47 -> 948f7ef2f
Add a simple snippet for a composite transform. Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/06f4ef40 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/06f4ef40 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/06f4ef40 Branch: refs/heads/master Commit: 06f4ef40bec7d6d3c5826fb9e0c164eb9c05efe7 Parents: 674ae9a Author: Ahmet Altay <[email protected]> Authored: Thu Apr 27 17:14:41 2017 -0700 Committer: Ahmet Altay <[email protected]> Committed: Thu Apr 27 17:35:34 2017 -0700 ---------------------------------------------------------------------- .../examples/snippets/snippets_test.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/06f4ef40/sdks/python/apache_beam/examples/snippets/snippets_test.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/examples/snippets/snippets_test.py b/sdks/python/apache_beam/examples/snippets/snippets_test.py index b8054ad..a3cdb24 100644 --- a/sdks/python/apache_beam/examples/snippets/snippets_test.py +++ b/sdks/python/apache_beam/examples/snippets/snippets_test.py @@ -903,6 +903,24 @@ class CombineTest(unittest.TestCase): p.run() +class PTransformTest(unittest.TestCase): + """Tests for PTransform.""" + + def test_composite(self): + + # [START model_composite_transform] + class ComputeWordLengths(beam.PTransform): + def expand(self, pcoll): + # transform logic goes here + return pcoll | beam.Map(lambda x: len(x)) + # [END model_composite_transform] + + p = TestPipeline() + lengths = p | beam.Create(["a", "ab", "abc"]) | ComputeWordLengths() + beam.assert_that(lengths, beam.equal_to([1, 2, 3])) + p.run() + + if __name__ == '__main__': logging.getLogger().setLevel(logging.INFO) unittest.main()
