tests

Author: DvirDukhan

tests/flow/tests_common.py

@@ -332,3 +332,8 @@ def test_lua_multi(env):
         exception = e
         env.assertEqual(type(exception), redis.exceptions.ResponseError)
         env.assertEqual("Cannot run RedisAI command within a transaction or a LUA script", exception.__str__())
+
+def test_info(env):
+    con = env.getConnection()
+    ret = con.execute_command('AI.INFO')
+    env.assertEqual(6, len(ret))

tests/flow/tests_onnx.py

@@ -393,3 +393,23 @@ def test_onnx_model_rdb_save_load(env):
     # Assert input model binary is equal to loaded model binary
     env.assertTrue(model_pb == model_serialized_after_rdbload)
 
+def tests_onnx_info(env):
+    if not TEST_ONNX:
+        env.debugPrint("skipping {} since TEST_ONNX=0".format(sys._getframe().f_code.co_name), force=True)
+        return
+    con = env.getConnection()
+
+    ret = con.execute_command('AI.INFO')
+    env.assertEqual(6, len(ret))
+
+    test_data_path = os.path.join(os.path.dirname(__file__), 'test_data')
+    linear_model_filename = os.path.join(test_data_path, 'linear_iris.onnx')
+
+    with open(linear_model_filename, 'rb') as f:
+        model_pb = f.read()
+
+    con.execute_command('AI.MODELSET', 'linear{1}', 'ONNX', DEVICE, 'BLOB', model_pb)
+    
+    ret = con.execute_command('AI.INFO')
+    env.assertEqual(8, len(ret))
+    env.assertEqual(b'ONNX version', ret[6])

tests/flow/tests_tensorflow.py

@@ -860,3 +860,23 @@ def test_tensorflow_modelrun_scriptrun_resnet(env):
     ret = con.execute_command('AI.TENSORGET', output_key, 'VALUES' )
     # tf model has 100 classes [0,999]
     env.assertEqual(ret[0]>=0 and ret[0]<1001, True)
+
+@skip_if_no_TF
+def test_tf_info(env):
+    con = env.getConnection()
+
+    ret = con.execute_command('AI.INFO')
+    env.assertEqual(6, len(ret))
+
+    test_data_path = os.path.join(os.path.dirname(__file__), 'test_data')
+    model_filename = os.path.join(test_data_path, 'graph.pb')
+
+    with open(model_filename, 'rb') as f:
+        model_pb = f.read()
+
+    con.execute_command('AI.MODELSET', 'm{1}', 'TF', DEVICE,
+                              'INPUTS', 'a', 'b', 'OUTPUTS', 'mul', 'BLOB', model_pb)
+    
+    ret = con.execute_command('AI.INFO')
+    env.assertEqual(8, len(ret))
+    env.assertEqual(b'TensorFlow version', ret[6])