usearch 0.11.0 → 0.12.0

package/src/test.cpp

@@ -2,7 +2,7 @@
  * @brief A trivial test.
  */
 #include <algorithm>
-#include <cassert>
+#include <stdexcept>
 
 #include "punned.hpp"
 
@@ -11,13 +11,20 @@ using namespace unum;
 
 using point_id_t = std::int64_t;
 
+void expect(bool must_be_true) {
+    if (!must_be_true)
+        throw std::runtime_error("Failed!");
+}
+
 template <typename scalar_at, typename index_at> void test3d(index_at&& index) {
-    using view_t = span_gt<scalar_at const>;
+
+    using scalar_t = scalar_at;
+    using view_t = span_gt<scalar_t const>;
     using distance_t = typename index_at::distance_t;
 
-    scalar_at vec[3] = {10, 20, 15};
-    scalar_at vec_a[3] = {15, 16, 17};
-    scalar_at vec_b[3] = {16, 17, 18};
+    scalar_t vec[3] = {10, 20, 15};
+    scalar_t vec_a[3] = {15, 16, 17};
+    scalar_t vec_b[3] = {16, 17, 18};
 
     index.reserve(10);
     index.add(42, view_t{&vec[0], 3ul});
@@ -27,16 +34,16 @@ template <typename scalar_at, typename index_at> void test3d(index_at&& index) {
     distance_t matched_distances[10] = {0};
     std::size_t matched_count = index.search(view_t{&vec[0], 3ul}, 5).dump_to(matched_labels, matched_distances);
 
-    assert(matched_count == 1);
-    assert(matched_labels[0] == 42);
-    assert(std::abs(matched_distances[0]) < 0.01);
+    expect(matched_count == 1);
+    expect(matched_labels[0] == 42);
+    expect(std::abs(matched_distances[0]) < 0.01);
 
     // Add more entries
     search_config_t search_config;
     search_config.exact = true;
     index.add(43, view_t{&vec_a[0], 3ul});
     index.add(44, view_t{&vec_b[0], 3ul});
-    assert(index.size() == 3);
+    expect(index.size() == 3);
 
     // Perform exact search
     matched_count = index.search(view_t{&vec[0], 3ul}, 5, search_config).dump_to(matched_labels, matched_distances);
@@ -45,47 +52,84 @@ template <typename scalar_at, typename index_at> void test3d(index_at&& index) {
     std::size_t count = 0;
     for (auto member : index) {
         point_id_t id = member.label;
-        assert(id >= 42 && id <= 44);
+        expect(id >= 42 && id <= 44);
         count++;
     }
-    assert((count == 3));
-    assert((index.stats(0).nodes == 3));
+    expect((count == 3));
+    expect((index.stats(0).nodes == 3));
 
     // Search again over reconstructed index
     index.save("tmp.usearch");
     index.load("tmp.usearch");
     matched_count = index.search(view_t{&vec[0], 3ul}, 5).dump_to(matched_labels, matched_distances);
-    assert(matched_count == 3);
-    assert(matched_labels[0] == 42);
-    assert(std::abs(matched_distances[0]) < 0.01);
+    expect(matched_count == 3);
+    expect(matched_labels[0] == 42);
+    expect(std::abs(matched_distances[0]) < 0.01);
 
     // Search again over mapped index
     index.view("tmp.usearch");
     matched_count = index.search(view_t{&vec[0], 3ul}, 5).dump_to(matched_labels, matched_distances);
-    assert(matched_count == 3);
-    assert(matched_labels[0] == 42);
-    assert(std::abs(matched_distances[0]) < 0.01);
+    expect(matched_count == 3);
+    expect(matched_labels[0] == 42);
+    expect(std::abs(matched_distances[0]) < 0.01);
 
-    assert(index.memory_usage() > 0);
-    assert(index.stats().max_edges > 0);
+    expect(index.memory_usage() > 0);
+    expect(index.stats().max_edges > 0);
 }
 
 template <typename scalar_at, typename index_at> void test3d_punned(index_at&& index) {
-    using view_t = span_gt<scalar_at const>;
+
+    using scalar_t = scalar_at;
+    using view_t = span_gt<scalar_t const>;
     using span_t = span_gt<scalar_at>;
-    using distance_t = typename index_at::distance_t;
 
-    scalar_at vec[3] = {10, 20, 15};
+    scalar_t vec[3] = {10, 20, 15};
 
     index.reserve(10);
     index.add(42, view_t{&vec[0], 3ul});
 
     // Reconstruct
-    scalar_at vec_reconstructed[3] = {0, 0, 0};
+    scalar_t vec_reconstructed[3] = {0, 0, 0};
     index.reconstruct(42, span_t{&vec_reconstructed[0], 3ul});
-    assert(vec_reconstructed[0] == vec[0]);
-    assert(vec_reconstructed[1] == vec[1]);
-    assert(vec_reconstructed[2] == vec[2]);
+    expect(vec_reconstructed[0] == vec[0]);
+    expect(vec_reconstructed[1] == vec[1]);
+    expect(vec_reconstructed[2] == vec[2]);
+}
+
+template <typename index_at> void test_sets(index_at&& index) {
+
+    using index_t = typename std::remove_reference<index_at>::type;
+    using scalar_t = typename index_t::scalar_t;
+    using view_t = span_gt<scalar_t const>;
+
+    scalar_t vec0[] = {10, 20};
+    scalar_t vec1[] = {10, 15, 20};
+    scalar_t vec2[] = {10, 20, 30, 35};
+
+    index.reserve(10);
+    index.add(42, view_t{&vec0[0], 2ul});
+    index.add(43, view_t{&vec1[0], 3ul});
+    index.add(44, view_t{&vec2[0], 4ul});
+
+    expect(index.size() == 3);
+}
+
+template <typename index_at> void test_sets_moved() {
+    {
+        index_at index;
+        test_sets(index);
+    }
+    {
+        index_at index;
+        index.reserve(1);
+        test_sets(index_at(std::move(index)));
+    }
+    {
+        index_at index;
+        index.reserve(1);
+        index_at index_moved = std::move(index);
+        test_sets(index_moved);
+    }
 }
 
 int main(int, char**) {
@@ -105,5 +149,11 @@ int main(int, char**) {
     test3d_punned<float>(punned_small_t::cos(3));
     test3d_punned<float>(punned_small_t::l2sq(3));
 
+    test_sets(index_gt<jaccard_gt<std::int32_t, float>, point_id_t, std::uint32_t, std::int32_t>{});
+    test_sets(index_gt<jaccard_gt<std::int64_t, float>, point_id_t, std::uint32_t, std::int64_t>{});
+
+    test_sets_moved<index_gt<jaccard_gt<std::int32_t, float>, point_id_t, std::uint32_t, std::int32_t>>();
+    test_sets_moved<index_gt<jaccard_gt<std::int64_t, float>, point_id_t, std::uint32_t, std::int64_t>>();
+
     return 0;
 }

package/wolfram/lib.cpp

@@ -10,23 +10,22 @@ EXTERN_C DLLEXPORT int WolframLibrary_initialize(WolframLibraryData libData) { r
 EXTERN_C DLLEXPORT void WolframLibrary_uninitialize(WolframLibraryData libData) { return; }
 
 EXTERN_C DLLEXPORT int IndexCreate(WolframLibraryData libData, mint Argc, MArgument* Args, MArgument Res) {
-    config_t config;
+    index_config_t config;
     char* accuracy_cstr = nullptr;
     char* metric_cstr = nullptr;
     try {
         accuracy_cstr = MArgument_getUTF8String(Args[1]);
         metric_cstr = MArgument_getUTF8String(Args[0]);
-        long dimensions = static_cast<std::size_t>(MArgument_getInteger(Args[2]));
-        config.max_elements = static_cast<std::size_t>(MArgument_getInteger(Args[3]));
+        std::size_t dimensions = static_cast<std::size_t>(MArgument_getInteger(Args[2]));
+        std::size_t capacity = static_cast<std::size_t>(MArgument_getInteger(Args[3]));
         config.connectivity = static_cast<std::size_t>(MArgument_getInteger(Args[4]));
         config.expansion_add = static_cast<std::size_t>(MArgument_getInteger(Args[5]));
         config.expansion_search = static_cast<std::size_t>(MArgument_getInteger(Args[6]));
-        config.max_threads_add = 4;
-        config.max_threads_search = 4;
 
-        accuracy_t accuracy = accuracy_from_name(accuracy_cstr, strlen(accuracy_cstr));
-        punned_t index = index_from_name<punned_t>( //
-            metric_cstr, strlen(metric_cstr), dimensions, accuracy, config);
+        accuracy_t accuracy = accuracy_from_name(accuracy_cstr, std::strlen(accuracy_cstr));
+        common_metric_kind_t metric_kind = common_metric_from_name(metric_cstr, std::strlen(metric_cstr));
+        punned_t index = make_punned<punned_t>(metric_kind, dimensions, accuracy, config);
+        index.reserve(capacity);
 
         punned_t* result_ptr = new punned_t(std::move(index));
         MArgument_setInteger(Res, (long)result_ptr);

package/Dockerfile

@@ -1,16 +0,0 @@
-# syntax=docker/dockerfile:1
-
-FROM ubuntu:23.04
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends python3.11 python3-pip && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
-RUN pip3 install --no-cache-dir --break-system-packages ucall usearch
-
-WORKDIR /usearch
-COPY python/usearch/server.py server.py
-
-ENTRYPOINT ["python3", "./server.py"]
-EXPOSE 8545

package/VERSION

@@ -1 +0,0 @@
-0.11.0