Skip to content

Commit

Permalink
Merge pull request #32530 from apache/users/damccorm/assert
Browse files Browse the repository at this point in the history
Move assert into pipeline
  • Loading branch information
liferoad authored Sep 22, 2024
2 parents 6adb145 + 149f355 commit 7c382df
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 12 deletions.
2 changes: 1 addition & 1 deletion .github/trigger_files/beam_PostCommit_Python.json
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"comment": "Modify this file in a trivial way to cause this test suite to run.",
"comment": "modify this file in a trivial way to cause this test suite to run.",
"modification": 1
}

23 changes: 12 additions & 11 deletions sdks/python/apache_beam/ml/gcp/visionml_test_it.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,17 +64,18 @@ def test_text_detection_with_language_hint(self):
context_side_input=beam.pvalue.AsDict(contexts))
| beam.ParDo(extract))

assert_that(
output,
equal_to([
'WAITING?\nPLEASE\nTURN OFF\nYOUR\nENGINE',
'WAITING?',
'PLEASE',
'TURN',
'OFF',
'YOUR',
'ENGINE'
]))
assert_that(
output,
equal_to([
'WAITING?\nPLEASE\nTURN OFF\nYOUR\nENGINE',
'WAITING',
'?',
'PLEASE',
'TURN',
'OFF',
'YOUR',
'ENGINE'
]))


if __name__ == '__main__':
Expand Down

0 comments on commit 7c382df

Please sign in to comment.