Send the path of test module .so to tests.sh and use it in tests_llapi.py to load the module.

Author: alonre24

opt/Makefile

@@ -276,6 +276,7 @@ test: build
 	$(SHOW)\
 		DEVICE=$(DEVICE) \
 		MODULE=$(realpath $(INSTALLED_TARGET)) \
+		TESTMOD=$(realpath $(BINDIR)/tests/module/testmod.so) \
 		CLUSTER=$(CLUSTER) \
 		GEN=$(GEN) AOF=$(AOF) SLAVES=$(SLAVES) \
 		VALGRIND=$(VALGRIND) \

tests/flow/includes.py

@@ -15,9 +15,7 @@
 import paella
 
 ROOT = os.environ.get("ROOT", None)
-if ROOT is None:
-    sys.stderr.write("ROOT was not defined in the environment.\n")
-    sys.exit(1)
+TESTMOD_PATH = os.environ.get("TESTMOD", None)
 MAX_ITERATIONS = 2 if os.environ.get("MAX_ITERATIONS") == None else os.environ.get("MAX_ITERATIONS")
 TEST_TF = os.environ.get("TEST_TF") != "0" and os.environ.get("WITH_TF") != "0"
 TEST_TFLITE = os.environ.get("TEST_TFLITE") != "0" and os.environ.get("WITH_TFLITE") != "0"

tests/flow/tests.sh

@@ -28,6 +28,9 @@ help() {
 		Argument variables:
 		VERBOSE=1     Print commands
 		IGNERR=1      Do not abort on error
+
+		MODULE=path     Path to redisai.so
+		TESTMOD=path    Path to LLAPI module
 		
 		DEVICE=cpu|gpu  Device for testing
 		GEN=0|1         General tests
@@ -115,6 +118,8 @@ OP=""
 
 MODULE=${MODULE:-$1}
 [[ -z $MODULE || ! -f $MODULE ]] && { echo "Module not found at ${MODULE}. Aborting."; exit 1; }
+TESTMOD=${TESTMOD}
+echo "Test module path is ${TESTMOD}"
 
 [[ $VALGRIND == 1 || $VGD == 1 ]] && valgrind_config
 

tests/flow/tests_llapi.py

@@ -8,18 +8,15 @@
 python -m RLTest --test tests_llapi.py --module path/to/redisai.so
 '''
 
-
-def ensure_test_module_loaded(f):
+def with_test_module(f):
     @wraps(f)
     def wrapper(env, *args, **kwargs):
-        #goal_dir = os.path.join(os.path.dirname(__file__), "../module/testmod.so")
-        TEST_MODULE_PATH = "{ROOT}/bin/linux-x64-release/src/tests/module/testmod.so".format(ROOT=ROOT)
         con = env.getConnection()
         modules = con.execute_command("MODULE", "LIST")
         if b'RAI_llapi' in [module[1] for module in modules]:
             return f(env, *args, **kwargs)
         try:
-            ret = con.execute_command('MODULE', 'LOAD', TEST_MODULE_PATH)
+            ret = con.execute_command('MODULE', 'LOAD', TESTMOD_PATH)
             env.assertEqual(ret, b'OK')
         except Exception as e:
             env.assertFalse(True)
@@ -29,15 +26,15 @@ def wrapper(env, *args, **kwargs):
     return wrapper
 
 
-@ensure_test_module_loaded
+@with_test_module
 def test_basic_check(env):
 
     con = env.getConnection()
     ret = con.execute_command("RAI_llapi.basic_check")
     env.assertEqual(ret, b'OK')
 
 
-@ensure_test_module_loaded
+@with_test_module
 def test_model_run_async(env):
 
     con = env.getConnection()
@@ -56,7 +53,7 @@ def test_model_run_async(env):
     env.assertEqual(ret, b'Async run success')
 
 
-@ensure_test_module_loaded
+@with_test_module
 def test_script_run_async(env):
 
     con = env.getConnection()
@@ -78,7 +75,7 @@ def test_script_run_async(env):
     env.assertEqual(ret, b'Async run success')
 
 
-@ensure_test_module_loaded
+@with_test_module
 def test_dag_build_and_run(env):
     con = env.getConnection()
 
@@ -105,7 +102,7 @@ def test_dag_build_and_run(env):
     env.assertEqual(ret, b'DAG run success')
 
 
-@ensure_test_module_loaded
+@with_test_module
 def test_dagrun_multidevice_resnet(env):
     con = env.getConnection()