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.')
def test_nested(self): with beam.Pipeline() as p: assert_that(p | FibTransform(6), equal_to([8]))