From e0bda54c4193f2222702b9490c52e157fd87fe1c 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 8d6e89d9..9c9a4a42 100644 --- a/tests/test_models/test_status_model.py +++ b/tests/test_models/test_status_model.py @@ -9,6 +9,8 @@ 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 1cc8d511..dfbfc1b1 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 @@ def test_save_with_update_fields_overrides_modified_provided_within_a(self) -> N 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[list | tuple | set] = ( [], # list (), # tuple set(), # set