Skip to content

Commit 20a9deb

Browse files
committed
Update test_example.py
Signed-off-by: xavier dupré <[email protected]>
1 parent 1d642f9 commit 20a9deb

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/test_example.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,14 @@ def run_example(self, name, expected=None):
2222
proc = subprocess.run(('python %s' % full).split(),
2323
capture_output=True, check=True)
2424
self.assertEqual(0, proc.returncode)
25+
out = proc.stdout.decode('ascii')
26+
if 'tensorflow_hub not installed' in out:
27+
return
28+
err = proc.stderr.decode('ascii')
29+
self.assertTrue(err is not None)
2530
if expected is not None:
26-
out = proc.stdout.decode('ascii')
2731
for exp in expected:
2832
self.assertIn(exp, out)
29-
err = proc.stderr.decode('ascii')
30-
self.assertTrue(err is not None)
3133

3234
@check_tf_min_version("2.3", "use tf.keras")
3335
@check_opset_min_version(12)

0 commit comments

Comments
 (0)