@@ -401,9 +401,9 @@ def test_dagexecute_modelexecute_multidevice_resnet(env):
401
401
'AI.SCRIPTEXECUTE' , script_name , 'wrong_fn' ,
402
402
'INPUTS' , 1 , image_key ,
403
403
'OUTPUTS' , 1 , temp_key1 )
404
- env .assertEquals (b'OK' , ret [0 ])
405
- env .assertEquals (type (ret [1 ]), redis .exceptions .ResponseError )
406
- env .assertEquals ("Function does not exist: wrong_fn" , ret [1 ].__str__ ())
404
+ env .assertEqual (b'OK' , ret [0 ])
405
+ env .assertEqual (type (ret [1 ]), redis .exceptions .ResponseError )
406
+ env .assertEqual ("Function does not exist: wrong_fn" , ret [1 ].__str__ ())
407
407
408
408
check_error_message (env , con , "Number of keys given as INPUTS here does not match model definition" ,
409
409
'AI.DAGEXECUTE' , 'ROUTING' , '{1}' ,
@@ -581,11 +581,11 @@ def test_dagexecute_modelexecute_multidevice_resnet_ensemble_alias(env):
581
581
'INPUTS' , 1 , temp_key1 ,
582
582
'OUTPUTS' , 1 , class_key_0 )
583
583
584
- env .assertEquals (b'OK' , ret [0 ])
585
- env .assertEquals (b'OK' , ret [1 ])
586
- env .assertEquals (b'OK' , ret [2 ])
587
- env .assertEquals (b'OK' , ret [3 ])
588
- env .assertEquals (b'NA' , ret [5 ])
584
+ env .assertEqual (b'OK' , ret [0 ])
585
+ env .assertEqual (b'OK' , ret [1 ])
586
+ env .assertEqual (b'OK' , ret [2 ])
587
+ env .assertEqual (b'OK' , ret [3 ])
588
+ env .assertEqual (b'NA' , ret [5 ])
589
589
env .assertEqual (type (ret [4 ]), redis .exceptions .ResponseError )
590
590
env .assertTrue ("list index out of range" in ret [4 ].__str__ ())
591
591
0 commit comments