Skip to content

Commit c9330e1

Browse files
author
hhsecond
committed
argument name change and test fixes
1 parent a40520c commit c9330e1

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/test.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -63,13 +63,13 @@ def test_run_tf_model(self):
6363

6464
con = self.get_client()
6565
con.modelset('m', Backend.tf, Device.cpu, model_pb,
66-
input=['a', 'b'], output='mul')
66+
inputs=['a', 'b'], outputs='mul')
6767

6868
# wrong model
6969
self.assertRaises(ResponseError,
7070
con.modelset, 'm', Backend.tf, Device.cpu,
7171
wrong_model_pb,
72-
input=['a', 'b'], output='mul')
72+
inputs=['a', 'b'], outputs='mul')
7373
# missing inputs/outputs
7474
self.assertRaises(ValueError,
7575
con.modelset, 'm', Backend.tf, Device.cpu,
@@ -79,7 +79,7 @@ def test_run_tf_model(self):
7979
self.assertRaises(ResponseError,
8080
con.modelset, 'm', Backend.torch, Device.cpu,
8181
model_pb,
82-
input=['a', 'b'], output='mul')
82+
inputs=['a', 'b'], outputs='mul')
8383

8484
con.tensorset('a', Tensor.scalar(DType.float, 2, 3))
8585
con.tensorset('b', Tensor.scalar(DType.float, 2, 3))
@@ -99,8 +99,8 @@ def bar(a, b):
9999
con.tensorset('b', Tensor.scalar(DType.float, 2, 3))
100100
# try with bad arguments:
101101
self.assertRaises(ResponseError,
102-
con.scriptrun, 'ket', 'bar', input='a', output='c')
103-
con.scriptrun('ket', 'bar', input=['a', 'b'], output='c')
102+
con.scriptrun, 'ket', 'bar', inputs='a', outputs='c')
103+
con.scriptrun('ket', 'bar', inputs=['a', 'b'], outputs='c')
104104
tensor = con.tensorget('c', as_type=Tensor)
105105
self.assertEqual([4, 6], tensor.value)
106106

0 commit comments

Comments
 (0)