split tests into packages

tests/test_model_definition/test_columns.py

@@ -0,0 +1,61 @@
+import datetime
+
+import databases
+import pydantic
+import pytest
+import sqlalchemy
+
+import ormar
+from tests.settings import DATABASE_URL
+
+database = databases.Database(DATABASE_URL, force_rollback=True)
+metadata = sqlalchemy.MetaData()
+
+
+def time():
+    return datetime.datetime.now().time()
+
+
+class Example(ormar.Model):
+    class Meta:
+        tablename = "example"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=200, default="aaa")
+    created: datetime.datetime = ormar.DateTime(default=datetime.datetime.now)
+    created_day: datetime.date = ormar.Date(default=datetime.date.today)
+    created_time: datetime.time = ormar.Time(default=time)
+    description: str = ormar.Text(nullable=True)
+    value: float = ormar.Float(nullable=True)
+    data: pydantic.Json = ormar.JSON(default={})
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@pytest.mark.asyncio
+async def test_model_crud():
+    async with database:
+        example = Example()
+        await example.save()
+
+        await example.load()
+        assert example.created.year == datetime.datetime.now().year
+        assert example.created_day == datetime.date.today()
+        assert example.description is None
+        assert example.value is None
+        assert example.data == {}
+
+        await example.update(data={"foo": 123}, value=123.456)
+        await example.load()
+        assert example.value == 123.456
+        assert example.data == {"foo": 123}
+
+        await example.delete()