Query: bubble up set_parameter_alias_vector_f32 #28

Author: Manthan Gupta

objectbox/c.py

@@ -696,7 +696,7 @@ def c_array_pointer(py_list: Union[List[Any], np.ndarray], c_type):
                                           ctypes.c_size_t])
 
 # OBX_C_API obx_err obx_query_param_alias_vector_float32(OBX_query* query, const char* alias, const float* value, size_t element_count);
-# TODO
+obx_query_param_alias_vector_float32 = c_fn_rc('obx_query_param_alias_vector_float32', [OBX_query_p, ctypes.c_char_p, ctypes.POINTER(ctypes.c_float), ctypes.c_size_t])
 
 # OBX_C_API obx_err obx_query_param_alias_string(OBX_query* query, const char* alias, const char* value);
 obx_query_param_alias_string = c_fn_rc('obx_query_param_alias_string', [OBX_query_p, ctypes.c_char_p, ctypes.c_char_p])

objectbox/query.py

@@ -135,3 +135,6 @@ def set_parameter_alias_string(self, alias: str, value: str):
 
     def set_parameter_alias_int(self, alias: str, value: int):
         return obx_query_param_alias_int(self._c_query, c_str(alias), value)
+
+    def set_parameter_alias_vector_f32(self, alias: str, value: Union[List[float], np.ndarray]):
+        return obx_query_param_alias_vector_float32(self._c_query, c_str(alias), c_array(value, ctypes.c_float), len(value))

tests/test_query.py

@@ -252,12 +252,19 @@ def test_set_parameter():
 
 
 def test_set_parameter_alias():
-    db = load_empty_test_objectbox()
+    db = create_test_objectbox()
     box = objectbox.Box(db, TestEntity)
 
     box.put(TestEntity(str="Foo", int64=2, int32=703, int8=101))
     box.put(TestEntity(str="FooBar", int64=10, int32=49, int8=45))
 
+    box_vector = objectbox.Box(db, VectorEntity)
+    box_vector.put(VectorEntity(name="Object 1", vector=[1, 1]))
+    box_vector.put(VectorEntity(name="Object 2", vector=[2, 2]))
+    box_vector.put(VectorEntity(name="Object 3", vector=[3, 3]))
+    box_vector.put(VectorEntity(name="Object 4", vector=[4, 4]))
+    box_vector.put(VectorEntity(name="Object 5", vector=[5, 5]))
+
     str_prop: Property = TestEntity.properties[1]
     qb = box.query(str_prop.equals("Foo").alias("foo_filter"))
 
@@ -281,3 +288,12 @@ def test_set_parameter_alias():
 
     assert query.count() == 2
 
+    vector_prop: Property = VectorEntity.get_property("vector")
+
+    query = box_vector.query(vector_prop.nearest_neighbor([3.4, 3.4], 3).alias("nearest_neighbour_filter")).build()
+    assert query.count() == 3
+    assert query.find_ids() == sorted([3, 4, 2])
+
+    query.set_parameter_alias_vector_f32("nearest_neighbour_filter", [4.9, 4.9])
+    assert query.count() == 3
+    assert query.find_ids() == sorted([5, 4, 3])