diff --git a/instant-distance-py/test/test.py b/instant-distance-py/test/test.py new file mode 100644 index 0000000..f93908f --- /dev/null +++ b/instant-distance-py/test/test.py @@ -0,0 +1,14 @@ +import instant_distance, random + +def main(): + points = [[random.random() for _ in range(300)] for _ in range(1024)] + config = instant_distance.Config() + (hnsw, ids) = instant_distance.Hnsw.build(points, config) + p = [random.random() for _ in range(300)] + search = instant_distance.Search() + hnsw.search(p, search) + for pid in search: + print(pid) + +if __name__ == '__main__': + main()