6
6
import os
7
7
import subprocess
8
8
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
10
10
11
11
12
12
class TestExample (unittest .TestCase ):
13
13
"""test examples"""
14
14
15
15
def run_example (self , name , expected = None ):
16
+ "Executes one example."
16
17
full = os .path .join (
17
18
os .path .abspath (os .path .dirname (__file__ )),
18
19
".." , "examples" , name )
19
20
if not os .path .exists (full ):
20
21
raise FileNotFoundError (full )
21
22
proc = subprocess .run (('python %s' % full ).split (),
22
- capture_output = True )
23
+ capture_output = True , check = True )
23
24
self .assertEqual (0 , proc .returncode )
24
25
if expected is not None :
25
26
out = proc .stdout .decode ('ascii' )
@@ -28,7 +29,7 @@ def run_example(self, name, expected=None):
28
29
err = proc .stderr .decode ('ascii' )
29
30
self .assertTrue (err is not None )
30
31
31
- @check_tf_min_version ("2.1 " , "use tf.keras" )
32
+ @check_tf_min_version ("2.0 " , "use tf.keras" )
32
33
@check_opset_min_version (12 )
33
34
@check_opset_max_version (13 )
34
35
def test_end2end_tfkeras (self ):
0 commit comments