Skip to content

Commit a20e73a

Browse files
committed
remove useless verbose argument from tests
1 parent 61953de commit a20e73a

File tree

2 files changed

+12
-15
lines changed

2 files changed

+12
-15
lines changed

tests/test_graph.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -95,17 +95,15 @@ class Tf2OnnxGraphTests(Tf2OnnxBackendTestBase):
9595

9696
def setUp(self):
9797
super().setUp()
98-
arg = namedtuple("Arg", "input inputs outputs verbose continue_on_error")
99-
self._args0 = arg(input="test", inputs=[], outputs=["output:0"],
100-
verbose=False, continue_on_error=False)
101-
self._args1 = arg(input="test", inputs=["input:0"], outputs=["output:0"],
102-
verbose=False, continue_on_error=False)
98+
arg = namedtuple("Arg", "input inputs outputs continue_on_error")
99+
self._args0 = arg(input="test", inputs=[], outputs=["output:0"], continue_on_error=False)
100+
self._args1 = arg(input="test", inputs=["input:0"], outputs=["output:0"], continue_on_error=False)
103101
self._args2 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output:0"],
104-
verbose=False, continue_on_error=False)
102+
continue_on_error=False)
105103
self._args3 = arg(input="test", inputs=["input1:0", "input2:0", "prob:0"], outputs=["output:0"],
106-
verbose=False, continue_on_error=False)
104+
continue_on_error=False)
107105
self._args4 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output1:0", "output2:0"],
108-
verbose=False, continue_on_error=False)
106+
continue_on_error=False)
109107

110108
def test_abs(self):
111109
with tf.Session() as sess:

tests/test_internals.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -52,13 +52,12 @@ def onnx_pretty(g, args=None):
5252
class Tf2OnnxInternalTests(Tf2OnnxBackendTestBase):
5353
def setUp(self):
5454
super().setUp()
55-
arg = namedtuple("Arg", "input inputs outputs verbose")
56-
self._args0 = arg(input="test", inputs=[], outputs=["output:0"], verbose=False)
57-
self._args1 = arg(input="test", inputs=["input:0"], outputs=["output:0"], verbose=False)
58-
self._args2 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output:0"], verbose=False)
59-
self._args3 = arg(input="test", inputs=["input1:0", "input2:0", "prob:0"], outputs=["output:0"], verbose=False)
60-
self._args4 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output1:0", "output2:0"],
61-
verbose=False)
55+
arg = namedtuple("Arg", "input inputs outputs")
56+
self._args0 = arg(input="test", inputs=[], outputs=["output:0"])
57+
self._args1 = arg(input="test", inputs=["input:0"], outputs=["output:0"])
58+
self._args2 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output:0"])
59+
self._args3 = arg(input="test", inputs=["input1:0", "input2:0", "prob:0"], outputs=["output:0"])
60+
self._args4 = arg(input="test", inputs=["input1:0", "input2:0"], outputs=["output1:0", "output2:0"])
6261

6362
@staticmethod
6463
def sample_net():

0 commit comments

Comments
 (0)