You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by al...@apache.org on 2017/04/28 00:35:46 UTC

[1/2] beam git commit: Add a simple snippet for a composite transform.

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 <al...@google.com>
Authored: Thu Apr 27 17:14:41 2017 -0700
Committer: Ahmet Altay <al...@google.com>
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()


[2/2] beam git commit: This closes #2756

Posted by al...@apache.org.
This closes #2756


Project: http://git-wip-us.apache.org/repos/asf/beam/repo
Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/948f7ef2
Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/948f7ef2
Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/948f7ef2

Branch: refs/heads/master
Commit: 948f7ef2fc37b30809cf1adb2631eeb6a1e5eff3
Parents: 674ae9a 06f4ef4
Author: Ahmet Altay <al...@google.com>
Authored: Thu Apr 27 17:35:37 2017 -0700
Committer: Ahmet Altay <al...@google.com>
Committed: Thu Apr 27 17:35:37 2017 -0700

----------------------------------------------------------------------
 .../examples/snippets/snippets_test.py            | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
----------------------------------------------------------------------