Exemplo n.º 1
0
 def test_unique_name(self):
   p = beam.Pipeline()
   _ = p | FibTransform(6)
   proto = p.to_runner_api()
   xforms = [x.unique_name for x in proto.components.transforms.values()]
   self.assertEqual(
       len(set(xforms)), len(xforms), msg='Transform names are not unique.')
   pcolls = [x.unique_name for x in proto.components.pcollections.values()]
   self.assertEqual(
       len(set(pcolls)), len(pcolls), msg='PCollection names are not unique.')
Exemplo n.º 2
0
 def test_nested(self):
     with beam.Pipeline() as p:
         assert_that(p | FibTransform(6), equal_to([8]))