tests: use objectbox API #56

Author: dan-obx

tests/common.py

@@ -2,11 +2,10 @@
 import pytest
 import objectbox
 from objectbox.logger import logger
-from objectbox.store import Store
 from tests.model import *
 import numpy as np
-from datetime import timezone
-
+from datetime import datetime, timezone
+from objectbox import *
 
 def remove_json_model_file():
     path = os.path.dirname(os.path.realpath(__file__))
@@ -16,7 +15,7 @@ def remove_json_model_file():
 
 
 def create_default_model():
-    model = objectbox.Model()
+    model = Model()
     model.entity(TestEntity)
     model.entity(TestEntityDatetime)
     model.entity(TestEntityFlex)

tests/model.py

@@ -1,6 +1,4 @@
-from objectbox.model import *
-from datetime import datetime
-
+from objectbox import *
 
 @Entity()
 class TestEntity:

tests/test_internals.py

@@ -1,5 +1,4 @@
 from objectbox import *
-from objectbox.model import *
 from objectbox.model.idsync import sync_model
 
 import os

tests/test_query.py

@@ -1,6 +1,5 @@
 import objectbox
 from objectbox import *
-from objectbox.model import *
 from objectbox.c import *
 from objectbox.query import *
 import pytest

tests/test_userclass.py

@@ -1,5 +1,4 @@
 from objectbox import *
-from objectbox.model import *
 from objectbox.model.idsync import sync_model
 
 def test_userclass():