small clean up

asg017 · Apr 7, 2023 · 1053147 · 1053147
1 parent 70b40bd
commit 1053147
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 17 deletions.
diff --git a/Makefile b/Makefile
@@ -104,7 +104,7 @@ test-python:
 
 test:
 	make test-loadable
-	#make test-python
+	make test-python
 
 
 test-loadable-3.41.0:

diff --git a/tests/np.py b/tests/np.py
@@ -0,0 +1,20 @@
+# https://github.com/facebookresearch/faiss/blob/74ee67aefc21859e02388f32f4876a0cbd21cabd/contrib/vecs_io.py#L20
+
+import numpy as np
+
+def ivecs_read(fname):
+    a = np.fromfile(fname, dtype='int32')
+    d = a[0]
+    return a.reshape(-1, d + 1)[:, 1:].copy()
+
+
+def fvecs_read(fname):
+    return ivecs_read(fname).view('float32')
+
+if __name__ == "__main__":
+  fv = fvecs_read('tests/data/siftsmall/siftsmall_base.fvecs')
+  print(fv.shape)
+  print(fv[0])
+  print(fv[1])
+  for i, row in enumerate(fv[:10]):
+    print(i, sum(row))
diff --git a/tests/test-sqlite-vector.sql b/tests/test-sqlite-vector.sql