diff --git a/tests/test_models/test_status_model.py b/tests/test_models/test_status_model.py index 8d6e89d..9c9a4a4 100644 --- a/tests/test_models/test_status_model.py +++ b/tests/test_models/test_status_model.py @@ -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 diff --git a/tests/test_models/test_timestamped_model.py b/tests/test_models/test_timestamped_model.py index 1cc8d51..6697941 100644 --- a/tests/test_models/test_timestamped_model.py +++ b/tests/test_models/test_timestamped_model.py @@ -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