diff options
| author | Mitja Felicijan <mitja.felicijan@gmail.com> | 2026-01-21 22:52:54 +0100 |
|---|---|---|
| committer | Mitja Felicijan <mitja.felicijan@gmail.com> | 2026-01-21 22:52:54 +0100 |
| commit | dcacc00e3750300617ba6e16eb346713f91a783a (patch) | |
| tree | 38e2d4fb5ed9d119711d4295c6eda4b014af73fd /examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py | |
| parent | 58dac10aeb8f5a041c46bddbeaf4c7966a99b998 (diff) | |
| download | crep-dcacc00e3750300617ba6e16eb346713f91a783a.tar.gz | |
Remove testing data
Diffstat (limited to 'examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py')
| -rw-r--r-- | examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py | 35 |
1 files changed, 0 insertions, 35 deletions
diff --git a/examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py b/examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py deleted file mode 100644 index 11c3c9b..0000000 --- a/examples/redis-unstable/modules/vector-sets/tests/basic_similarity.py +++ /dev/null @@ -1,35 +0,0 @@ -from test import TestCase - -class BasicSimilarity(TestCase): - def getname(self): - return "VSIM reported distance makes sense with 4D vectors" - - def test(self): - # Add two very similar vectors, one different - vec1 = [1, 0, 0, 0] - vec2 = [0.99, 0.01, 0, 0] - vec3 = [0.1, 1, -1, 0.5] - - # Add vectors using VALUES format - self.redis.execute_command('VADD', self.test_key, 'VALUES', 4, - *[str(x) for x in vec1], f'{self.test_key}:item:1') - self.redis.execute_command('VADD', self.test_key, 'VALUES', 4, - *[str(x) for x in vec2], f'{self.test_key}:item:2') - self.redis.execute_command('VADD', self.test_key, 'VALUES', 4, - *[str(x) for x in vec3], f'{self.test_key}:item:3') - - # Query similarity with vec1 - result = self.redis.execute_command('VSIM', self.test_key, 'VALUES', 4, - *[str(x) for x in vec1], 'WITHSCORES') - - # Convert results to dictionary - results_dict = {} - for i in range(0, len(result), 2): - key = result[i].decode() - score = float(result[i+1]) - results_dict[key] = score - - # Verify results - assert results_dict[f'{self.test_key}:item:1'] > 0.99, "Self-similarity should be very high" - assert results_dict[f'{self.test_key}:item:2'] > 0.99, "Similar vector should have high similarity" - assert results_dict[f'{self.test_key}:item:3'] < 0.8, "Not very similar vector should have low similarity" |
