Skip to content

Commit 93fbce4

Browse files
committed
Test tensor create from blob.
1 parent 1c677ea commit 93fbce4

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/flow/tests_withGears.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,8 +336,15 @@ def TensorCreate_FromValues(record):
336336
tensor = redisAI.createTensorFromValues('DOUBLE', [2,2], [1.0, 2.0, 3.0, 4.0])
337337
redisAI.setTensorInKey('test1_res{1}', tensor)
338338
return "test1_OK"
339+
340+
def TensorCreate_FromBlob(record):
341+
tensor_blob = bytearray([5, 6, 7, 8])
342+
tensor = redisAI.createTensorFromBlob('INT8', [2,2], tensor_blob)
343+
redisAI.setTensorInKey('test2_res{1}', tensor)
344+
return "test2_OK"
339345
340346
GB("CommandReader").map(TensorCreate_FromValues).register(trigger="TensorCreate_FromValues_test1")
347+
GB("CommandReader").map(TensorCreate_FromBlob).register(trigger="TensorCreate_FromBlob_test2")
341348
'''
342349

343350
con = env.getConnection()
@@ -348,3 +355,9 @@ def TensorCreate_FromValues(record):
348355

349356
values = con.execute_command('AI.TENSORGET', 'test1_res{1}', 'VALUES')
350357
env.assertEqual(values, [b'1', b'2', b'3', b'4'])
358+
359+
ret = con.execute_command('rg.trigger', 'TensorCreate_FromBlob_test2')
360+
env.assertEqual(ret[0], b'test2_OK')
361+
362+
values = con.execute_command('AI.TENSORGET', 'test2_res{1}', 'VALUES')
363+
env.assertEqual(values, [5, 6, 7, 8])

0 commit comments

Comments
 (0)