Skip to content

Commit 3cf1354

Browse files
committed
fixed deprecated assert
1 parent 0dbe985 commit 3cf1354

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

tests/flow/includes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def ensureSlaveSynced(con, env, timeout_ms=0):
4545
env.debugPring(str(ex), force=True)
4646
env.assertFalse(True)
4747
return
48-
env.assertEquals(number_replicas, 1)
48+
env.assertEqual(number_replicas, 1)
4949

5050

5151
# Ensures command is sent and forced disconnect

tests/flow/tests_dag.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -401,9 +401,9 @@ def test_dagexecute_modelexecute_multidevice_resnet(env):
401401
'AI.SCRIPTEXECUTE', script_name, 'wrong_fn',
402402
'INPUTS', 1, image_key,
403403
'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__())
407407

408408
check_error_message(env, con, "Number of keys given as INPUTS here does not match model definition",
409409
'AI.DAGEXECUTE', 'ROUTING', '{1}',
@@ -581,11 +581,11 @@ def test_dagexecute_modelexecute_multidevice_resnet_ensemble_alias(env):
581581
'INPUTS', 1, temp_key1,
582582
'OUTPUTS', 1, class_key_0)
583583

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])
589589
env.assertEqual(type(ret[4]), redis.exceptions.ResponseError)
590590
env.assertTrue("list index out of range" in ret[4].__str__())
591591

0 commit comments

Comments
 (0)