Skip to content

Commit 47847f8

Browse files
committed
Fix test error
1 parent ae53b8e commit 47847f8

File tree

1 file changed

+17
-12
lines changed

1 file changed

+17
-12
lines changed

tests/fields/test_time.py

Lines changed: 17 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from tests import testmodels
1212
from tortoise import fields, timezone
1313
from tortoise.contrib import test
14-
from tortoise.contrib.test.condition import In, NotIn
14+
from tortoise.contrib.test.condition import NotIn
1515
from tortoise.exceptions import ConfigurationError, IntegrityError
1616
from tortoise.expressions import F
1717
from tortoise.timezone import UTC, get_default_timezone, parse_timezone
@@ -225,14 +225,15 @@ async def test_datetime_filter_by_year_month_day(db):
225225

226226

227227
# ============================================================================
228-
# TestTimeFields (sqlite/postgres) -> test_time_*
228+
# TestTimeFields (postgres) -> test_time_*
229229
# ============================================================================
230230

231231

232232
@pytest.mark.asyncio
233-
@test.requireCapability(dialect=In("sqlite", "postgres"))
233+
@test.requireCapability(dialect=NotIn("sqlite")) # 'datetime.time' is not supported by sqlite3
234+
@test.requireCapability(dialect="postgres")
234235
async def test_time_create(db):
235-
"""Test creating time fields (sqlite/postgres)."""
236+
"""Test creating time fields (postgres)."""
236237
model = testmodels.TimeFields
237238
now = timezone.now().timetz()
238239
obj0 = await model.create(time=now)
@@ -241,19 +242,21 @@ async def test_time_create(db):
241242

242243

243244
@pytest.mark.asyncio
244-
@test.requireCapability(dialect=In("sqlite", "postgres"))
245+
@test.requireCapability(dialect=NotIn("sqlite")) # 'datetime.time' is not supported by sqlite3
246+
@test.requireCapability(dialect="postgres")
245247
async def test_time_cast(db):
246-
"""Test time field accepts ISO format string (sqlite/postgres)."""
248+
"""Test time field accepts ISO format string (postgres)."""
247249
model = testmodels.TimeFields
248250
obj0 = await model.create(time="21:00+00:00")
249251
obj1 = await model.get(id=obj0.id)
250252
assert obj1.time == time.fromisoformat("21:00+00:00")
251253

252254

253255
@pytest.mark.asyncio
254-
@test.requireCapability(dialect=In("sqlite", "postgres"))
256+
@test.requireCapability(dialect=NotIn("sqlite")) # 'datetime.time' is not supported by sqlite3
257+
@test.requireCapability(dialect="postgres")
255258
async def test_time_values(db):
256-
"""Test time field in values() query (sqlite/postgres)."""
259+
"""Test time field in values() query (postgres)."""
257260
model = testmodels.TimeFields
258261
now = timezone.now().timetz()
259262
obj0 = await model.create(time=now)
@@ -262,9 +265,10 @@ async def test_time_values(db):
262265

263266

264267
@pytest.mark.asyncio
265-
@test.requireCapability(dialect=In("sqlite", "postgres"))
268+
@test.requireCapability(dialect=NotIn("sqlite")) # 'datetime.time' is not supported by sqlite3
269+
@test.requireCapability(dialect="postgres")
266270
async def test_time_values_list(db):
267-
"""Test time field in values_list() query (sqlite/postgres)."""
271+
"""Test time field in values_list() query (postgres)."""
268272
model = testmodels.TimeFields
269273
now = timezone.now().timetz()
270274
obj0 = await model.create(time=now)
@@ -273,9 +277,10 @@ async def test_time_values_list(db):
273277

274278

275279
@pytest.mark.asyncio
276-
@test.requireCapability(dialect=In("sqlite", "postgres"))
280+
@test.requireCapability(dialect=NotIn("sqlite")) # 'datetime.time' is not supported by sqlite3
281+
@test.requireCapability(dialect="postgres")
277282
async def test_time_get(db):
278-
"""Test getting by time field (sqlite/postgres)."""
283+
"""Test getting by time field (postgres)."""
279284
model = testmodels.TimeFields
280285
now = timezone.now().timetz()
281286
await model.create(time=now)

0 commit comments

Comments
 (0)