Merge pull request apache#1 from aaltay/labels

fix pipeline test
cosmoskitten · Jul 22, 2016 · df3511d · df3511d
2 parents f18f4f0 + a138ff2
commit df3511d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sdks/python/apache_beam/pipeline_test.py b/sdks/python/apache_beam/pipeline_test.py
@@ -183,7 +183,7 @@ def apply(self, pcoll):
 
     self.assertEqual(
         ['a-x', 'b-x', 'c-x'],
-        sorted(['a', 'b', 'c'] | '-x' >> AddSuffix()))
+        sorted(['a', 'b', 'c'] | 'AddSuffix' >> AddSuffix('-x')))
 
   def test_cached_pvalues_are_refcounted(self):
     """Test that cached PValues are refcounted and deleted.
@@ -218,7 +218,7 @@ def create_dupes(o, _):
         biglist
         | 'oom:addone' >> Map(lambda x: (x, 1))
         | 'oom:dupes' >> FlatMap(create_dupes,
-                  AsIter(biglist)).with_outputs('side', main='main'))
+            AsIter(biglist)).with_outputs('side', main='main'))
     result = (
         (dupes.side, dupes.main, dupes.side)
         | 'oom:flatten' >> Flatten()
@@ -232,8 +232,8 @@ def create_dupes(o, _):
         {
             'oom:flatten': 3 * num_elements,
             ('oom:combine/GroupByKey/reify_windows', None): 3 * num_elements,
-            ('oom:dupes/oom:dupes', 'side'): num_elements,
-            ('oom:dupes/oom:dupes', None): num_elements,
+            ('oom:dupes/FlatMap(create_dupes)', 'side'): num_elements,
+            ('oom:dupes/FlatMap(create_dupes)', None): num_elements,
             'oom:create': num_elements,
             ('oom:addone', None): num_elements,
             'oom:combine/GroupByKey/group_by_key': 1,