properties: nearest_neighbor bound to Float32Vector exclusively; tests: added bool #29

Author: dan-obx

objectbox/model/properties.py

@@ -320,9 +320,6 @@ def contains_key_value(self, key: str, value: str, case_sensitive: bool = True)
 class _VectorProperty(Property):
     def __init__(self, py_type : Type, **kwargs):
         super(_VectorProperty, self).__init__(py_type, **kwargs)
-    def nearest_neighbor(self, query_vector, element_count: int) -> PropertyQueryCondition:
-        args = {'query_vector': query_vector, 'element_count': element_count}
-        return PropertyQueryCondition(self._id, PropertyQueryConditionOp.NEAREST_NEIGHBOR, args)
 
 class BoolVector(_VectorProperty):
     def __init__(self, id: int = 0, uid: int = 0, **kwargs):
@@ -350,6 +347,9 @@ def __init__(self, id: int = 0, uid: int = 0, **kwargs):
 class Float32Vector(_VectorProperty):
     def __init__(self, id: int = 0, uid: int = 0, **kwargs):
         super(Float32Vector, self).__init__(np.ndarray, type=PropertyType.floatVector, id=id, uid=uid, **kwargs)
+    def nearest_neighbor(self, query_vector, element_count: int) -> PropertyQueryCondition:
+        args = {'query_vector': query_vector, 'element_count': element_count}
+        return PropertyQueryCondition(self._id, PropertyQueryConditionOp.NEAREST_NEIGHBOR, args)
 
 class Float64Vector(_VectorProperty):
     def __init__(self, id: int = 0, uid: int = 0, **kwargs):

tests/test_query.py

@@ -12,14 +12,24 @@ def test_basics():
     store = create_test_store()
 
     box_test_entity = store.box(TestEntity)
-    box_test_entity.put(TestEntity(str="foo", int64=123))
-    box_test_entity.put(TestEntity(str="bar", int64=456))
+    id1 = box_test_entity.put(TestEntity(bool=True, str="foo", int64=123))
+    id2 = box_test_entity.put(TestEntity(bool=False, str="bar", int64=456))
 
     box_vector_entity = store.box(VectorEntity)
     box_vector_entity.put(VectorEntity(name="Object 1", vector_euclidean=[1, 1]))
     box_vector_entity.put(VectorEntity(name="Object 2", vector_euclidean=[2, 2]))
     box_vector_entity.put(VectorEntity(name="Object 3", vector_euclidean=[3, 3]))
 
+    # Bool query
+    bool_prop: Property = TestEntity.get_property("bool")
+    query = box_test_entity.query(bool_prop.equals(True)).build()
+    assert query.count() == 1
+    assert query.find()[0].id == id1
+    
+    query = box_test_entity.query(bool_prop.equals(False)).build()
+    assert query.count() == 1
+    assert query.find()[0].id == id2
+
     # String query
     str_prop: Property = TestEntity.get_property("str")
 
@@ -194,7 +204,8 @@ def test_float_scalars():
     box_test_entity = store.box(TestEntity)
     id1 = box_test_entity.put(TestEntity(float32=12, float64=12))
     id2 = box_test_entity.put(TestEntity(float32=45, float64=45))
-   
+  
+    # Test int scalar literals
     props = [ "float32", "float64" ]
     for p in props:
         prop = TestEntity.get_property(p)
@@ -217,7 +228,7 @@ def test_float_scalars():
         assert query.find()[0].id == id1
         assert query.find()[1].id == id2
 
-    # Test float scalar values 
+    # Test float scalar literals 
     for p in props:
         prop = TestEntity.get_property(p)
         query = box_test_entity.query(prop.greater_or_equal(11.0)).build()