def testNoIncludeOrExcludeFilters(self):
with beam.Pipeline() as pipeline:
got = (pipeline
| 'Create' >> beam.Create([{
constants.INPUT_KEY: 'input',
'other': 2
}])
| 'EvaluateExtracts' >>
analysis_table_evaluator.EvaluateExtracts())
def check_result(got):
try:
self.assertEqual(got, [{'other': 2}])
except AssertionError as err:
raise util.BeamAssertException(err)
util.assert_that(got[constants.ANALYSIS_KEY], check_result)
def testExcludeFilter(self):
with beam.Pipeline() as pipeline:
got = (pipeline
| 'Create' >> beam.Create([{
'a': 1,
'b': 2
}])
| 'EvaluateExtracts' >>
analysis_table_evaluator.EvaluateExtracts(exclude=['a']))
def check_result(got):
try:
self.assertEqual(got, [{'b': 2}])
except AssertionError as err:
raise util.BeamAssertException(err)
util.assert_that(got[constants.ANALYSIS_KEY], check_result)