From 949d110d043e8a38ee8d0127c5309174a7d8624a Mon Sep 17 00:00:00 2001 From: Maarten ter Huurne Date: Wed, 27 Mar 2024 18:32:59 +0100 Subject: [PATCH] Annotate the `test_models` package --- tests/test_models/test_status_model.py | 2 ++ tests/test_models/test_timestamped_model.py | 3 ++- 2 files changed, 4 insertions(+), 1 deletion(-) 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