Skip to content

Commit 653ff42

Browse files
author
DvirDukhan
committed
fixed tests messages
1 parent 4f51679 commit 653ff42

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/tests_dag.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def test_dag_scriptrun_errors(env):
270270
except Exception as e:
271271
exception = e
272272
env.assertEqual(type(exception), redis.exceptions.ResponseError)
273-
env.assertEqual("ERR unsupported command within DAG",exception.__str__())
273+
env.assertEqual("function name not specified", exception.__str__())
274274

275275

276276
def test_dag_modelrun_financialNet_errors(env):
@@ -305,7 +305,7 @@ def test_dag_modelrun_financialNet_errors(env):
305305
except Exception as e:
306306
exception = e
307307
env.assertEqual(type(exception), redis.exceptions.ResponseError)
308-
env.assertEqual("ERR unsupported command within DAG",exception.__str__())
308+
env.assertEqual("Number of names given as INPUTS during MODELSET and keys given as INPUTS here do not match", exception.__str__())
309309

310310

311311
def test_dag_local_tensorset(env):

test/tests_tensorflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,7 @@ def test_run_tf_model_errors(env):
458458
except Exception as e:
459459
exception = e
460460
env.assertEqual(type(exception), redis.exceptions.ResponseError)
461-
env.assertEqual("tensor key is empty", exception.__str__())
461+
env.assertEqual("Number of names given as OUTPUTS during MODELSET and keys given as OUTPUTS here do not match", exception.__str__())
462462

463463
try:
464464
con.execute_command('AI.MODELRUN', 'm{1}', 'OUTPUTS', 'c{1}')

0 commit comments

Comments
 (0)