apunkt/sqlite-vec
1
0
Fork 0
mirror of https://github.com/asg017/sqlite-vec.git synced 2026-04-25 08:46:49 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Initial commit

Browse source
This commit is contained in:
Alex Garcia 2024-04-20 13:38:58 -07:00
commit 4c8ad629e0
28 changed files with 6758 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/utils.py Normal file
View file

@ -0,0 +1,22 @@
import numpy as np
from io import BytesIO
def to_npy(arr):
buf = BytesIO()
np.save(buf, arr)
buf.seek(0)
return buf.read()
to_npy(np.array([[1.0, 2.0, 3.0], [2.0, 3.0, 4.0]], dtype=np.float32))
print(to_npy(np.array([[1.0, 2.0]], dtype=np.float32)))
print(to_npy(np.array([1.0, 2.0], dtype=np.float32)))
to_npy(
np.array(
[np.zeros(10), np.zeros(10), np.zeros(10), np.zeros(10), np.zeros(10)],
dtype=np.float32,
)
)