Annotate the test_models package

This commit is contained in:
Maarten ter Huurne 2024-03-27 18:32:59 +01:00
parent e4c88103f5
commit 949d110d04
2 changed files with 4 additions and 1 deletions

View file

@ -9,6 +9,8 @@ from tests.models import CustomManagerStatusModel, Status, StatusPlainTuple
class StatusModelTests(TestCase):
model: type[Status] | type[StatusPlainTuple]
def setUp(self) -> None:
self.model = Status
self.on_hold = Status.STATUS.on_hold

View file

@ -1,5 +1,6 @@
from __future__ import annotations
from collections.abc import Iterable
from datetime import datetime, timedelta, timezone
import time_machine
@ -114,7 +115,7 @@ class TimeStampedModelTests(TestCase):
self.assertEqual(t1.modified, datetime(2020, 1, 2, tzinfo=timezone.utc))
def test_save_is_skipped_for_empty_update_fields_iterable(self) -> None:
tests = (
tests: Iterable[Iterable[str]] = (
[], # list
(), # tuple
set(), # set