PR fixes

Author: alonre24

tests/flow/tests_withGears.py

@@ -28,22 +28,6 @@ def wrapper(env, *args, **kwargs):
     return wrapper
 
 
-@skip_if_gears_not_loaded
-def test_ping_gears(env):
-
-    script = '''
-def ping(record):
-    return "pong"
-    
-GB("CommandReader").map(ping).register(trigger="ping_test")
-'''
-    con = env.getConnection()
-    ret = con.execute_command('rg.pyexecute', script)
-    env.assertEqual(ret, b'OK')
-    ret = con.execute_command('rg.trigger', 'ping_test')
-    env.assertEqual(ret[0], b'pong')
-
-
 @skip_if_gears_not_loaded
 def test_model_run(env):
     script = '''
@@ -56,7 +40,7 @@ def ModelRun_oldAPI(record):
     modelRunner = redisAI.createModelRunner('m{1}')
     redisAI.modelRunnerAddInput(modelRunner, 'a', tensors[0])
     redisAI.modelRunnerAddInput(modelRunner, 'b', tensors[1])
-    redisAI.modelRunnerAddOutput(modelRunner, 'mul')
+    redisAI.modelRunnerAddOutput(modelRunner, 'c')
     res = redisAI.modelRunnerRun(modelRunner)
     redisAI.setTensorInKey('c{1}', res[0])
     return "ModelRun_oldAPI_OK"
@@ -67,7 +51,7 @@ async def ModelRun_Async(record):
     modelRunner = redisAI.createModelRunner('m{1}')
     redisAI.modelRunnerAddInput(modelRunner, 'a', tensors[0])
     redisAI.modelRunnerAddInput(modelRunner, 'b', tensors[1])
-    redisAI.modelRunnerAddOutput(modelRunner, 'mul')
+    redisAI.modelRunnerAddOutput(modelRunner, 'c')
     res = await redisAI.modelRunnerRunAsync(modelRunner)
     redisAI.setTensorInKey('c{1}', res[0])
     return "ModelRun_Async_OK"
@@ -108,6 +92,8 @@ async def ModelRun_AsyncRunError(record):
     env.assertEqual(ret[0], b'ModelRun_oldAPI_OK')
     values = con.execute_command('AI.TENSORGET', 'c{1}', 'VALUES')
     env.assertEqual(values, [b'4', b'9', b'4', b'9'])
+    ret = con.execute_command('DEL', 'c{1}')
+    env.assertEqual(ret, 1)
 
     ret = con.execute_command('rg.trigger', 'ModelRun_Async_test2')
     env.assertEqual(ret[0], b'ModelRun_Async_OK')
@@ -186,6 +172,8 @@ async def ScriptRun_AsyncRunError(record):
     env.assertEqual(ret[0], b'ScriptRun_oldAPI_OK')
     values = con.execute_command('AI.TENSORGET', 'c{1}', 'VALUES')
     env.assertEqual(values, [b'4', b'6', b'4', b'6'])
+    ret = con.execute_command('DEL', 'c{1}')
+    env.assertEqual(ret, 1)
 
     ret = con.execute_command('rg.trigger', 'ScriptRun_Async_test2')
     env.assertEqual(ret[0], b'ScriptRun_Async_OK')
@@ -194,8 +182,7 @@ async def ScriptRun_AsyncRunError(record):
 
     ret = con.execute_command('rg.trigger', 'ScriptRun_AsyncRunError_test3')
     # This should raise an exception
-    error_string = b'attempted to get undefined function bad_func'
-    env.assertEqual(str(ret[0])[:len(error_string)+2]+"'", "{}".format(error_string))
+    env.assertTrue(str(ret[0]).startswith("b'attempted to get undefined function bad_func"))
 
 
 @skip_if_gears_not_loaded
@@ -314,9 +301,7 @@ async def DAGRun_addOpsFromString(record):
 
     ret = con.execute_command('rg.trigger', 'DAGRun_test4')
     # This should raise an exception
-
-    error_string = b'attempted to get undefined function no_func'
-    env.assertEqual(str(ret[0])[:len(error_string)+2]+"'", "{}".format(error_string))
+    env.assertTrue(str(ret[0]).startswith("b'attempted to get undefined function no_func"))
 
     ret = con.execute_command('rg.trigger', 'DAGRun_test5')
     env.assertEqual(ret[0], b'test5_OK')