mirror of
https://github.com/redis/redis.git
synced 2026-02-03 20:39:54 -05:00
78e0d87177
git-subtree-dir: modules/vector-sets git-subtree-mainline:8ea8f4220cgit-subtree-split:c6db0a7c20
27 lines
1 KiB
Python
27 lines
1 KiB
Python
from test import TestCase, generate_random_vector
|
|
import struct
|
|
|
|
class VREM_LastItemDeletesKey(TestCase):
|
|
def getname(self):
|
|
return "VREM last item deletes key"
|
|
|
|
def test(self):
|
|
# Generate a random vector
|
|
vec = generate_random_vector(4)
|
|
vec_bytes = struct.pack('4f', *vec)
|
|
|
|
# Add the vector to the key
|
|
result = self.redis.execute_command('VADD', self.test_key, 'FP32', vec_bytes, f'{self.test_key}:item:1')
|
|
assert result == 1, "VADD should return 1 for first item"
|
|
|
|
# Verify the key exists
|
|
exists = self.redis.exists(self.test_key)
|
|
assert exists == 1, "Key should exist after VADD"
|
|
|
|
# Remove the item
|
|
result = self.redis.execute_command('VREM', self.test_key, f'{self.test_key}:item:1')
|
|
assert result == 1, "VREM should return 1 for successful removal"
|
|
|
|
# Verify the key no longer exists
|
|
exists = self.redis.exists(self.test_key)
|
|
assert exists == 0, "Key should no longer exist after VREM of last item"
|