Skip to content

Commit d617b26

Browse files
committed
Annotate the test_models package
1 parent 1279c9f commit d617b26

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

tests/test_models/test_status_model.py

+2
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@
99

1010

1111
class StatusModelTests(TestCase):
12+
model: type[Status] | type[StatusPlainTuple]
13+
1214
def setUp(self) -> None:
1315
self.model = Status
1416
self.on_hold = Status.STATUS.on_hold

tests/test_models/test_timestamped_model.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from __future__ import annotations
22

3+
from collections.abc import Iterable
34
from datetime import datetime, timedelta, timezone
45

56
import time_machine
@@ -114,7 +115,7 @@ def test_save_with_update_fields_overrides_modified_provided_within_a(self) -> N
114115
self.assertEqual(t1.modified, datetime(2020, 1, 2, tzinfo=timezone.utc))
115116

116117
def test_save_is_skipped_for_empty_update_fields_iterable(self) -> None:
117-
tests = (
118+
tests: Iterable[list | tuple | set] = (
118119
[], # list
119120
(), # tuple
120121
set(), # set

0 commit comments

Comments
 (0)