Box.remove support both object and ID argument

Author: Ivan Dlugos

objectbox/box.py

@@ -116,7 +116,11 @@ def get_all(self) -> list:
         finally:
             obx_bytes_array_free(c_bytes_array_p)
 
-    def remove(self, id: int):
+    def remove(self, id_or_object):
+        if isinstance(id_or_object, self._entity.cls):
+            id = self._entity.get_object_id(id_or_object)
+        else:
+            id = id_or_object
         obx_box_remove(self._c_box, id)
 
     def remove_all(self) -> int:

tests/test_basics.py

@@ -85,12 +85,15 @@ def test_box_basics():
     assert_equal(read, object)
 
     # remove
-    box.remove(object.id)
+    box.remove(object)
+    box.remove(1)
 
-    # check it's missing
-    assert box.count() == 1
+    # check they're gone
+    assert box.count() == 0
     with pytest.raises(objectbox.NotFoundException):
         box.get(object.id)
+    with pytest.raises(objectbox.NotFoundException):
+        box.get(1)
 
 
 def test_box_bulk():