Skip to content

Commit 8eb9c28

Browse files
committed
Update test_example.py
Signed-off-by: xavier dupré <[email protected]>
1 parent 6ed70d2 commit 8eb9c28

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/test_example.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,21 @@
66
import os
77
import subprocess
88
import unittest
9-
from common import check_opset_min_version, check_opset_max_version
9+
from common import check_opset_min_version, check_opset_max_version, check_tf_min_version
1010

1111

1212
class TestExample(unittest.TestCase):
1313
"""test examples"""
1414

1515
def run_example(self, name, expected=None):
16+
"Executes one example."
1617
full = os.path.join(
1718
os.path.abspath(os.path.dirname(__file__)),
1819
"..", "examples", name)
1920
if not os.path.exists(full):
2021
raise FileNotFoundError(full)
2122
proc = subprocess.run(('python %s' % full).split(),
22-
capture_output=True)
23+
capture_output=True, check=True)
2324
self.assertEqual(0, proc.returncode)
2425
if expected is not None:
2526
out = proc.stdout.decode('ascii')
@@ -28,7 +29,7 @@ def run_example(self, name, expected=None):
2829
err = proc.stderr.decode('ascii')
2930
self.assertTrue(err is not None)
3031

31-
@check_tf_min_version("2.1", "use tf.keras")
32+
@check_tf_min_version("2.0", "use tf.keras")
3233
@check_opset_min_version(12)
3334
@check_opset_max_version(13)
3435
def test_end2end_tfkeras(self):

0 commit comments

Comments
 (0)