diff options
| author | Max Nanis | 2026-08-18 16:35:11 -0700 |
|---|---|---|
| committer | Max Nanis | 2026-08-18 16:35:11 -0700 |
| commit | 0742210fc7eaa54d8068bfd18414158dc9b33672 (patch) | |
| tree | 5f42b671ccbd9cc44f1f62229cac16a5e455317d /tests/models/thl/test_contest | |
| parent | 6b4f8d6b8700a65bdf74beb3d95c969a81464a56 (diff) | |
| download | generalresearch-0742210fc7eaa54d8068bfd18414158dc9b33672.tar.gz generalresearch-0742210fc7eaa54d8068bfd18414158dc9b33672.zip | |
Ruff cleanup
Diffstat (limited to 'tests/models/thl/test_contest')
| -rw-r--r-- | tests/models/thl/test_contest/test_contest.py | 10 | ||||
| -rw-r--r-- | tests/models/thl/test_contest/test_leaderboard_contest.py | 5 |
2 files changed, 10 insertions, 5 deletions
diff --git a/tests/models/thl/test_contest/test_contest.py b/tests/models/thl/test_contest/test_contest.py index d53eee5..0fbd4cc 100644 --- a/tests/models/thl/test_contest/test_contest.py +++ b/tests/models/thl/test_contest/test_contest.py @@ -1,4 +1,8 @@ +from typing import Callable + import pytest + +from generalresearch.models.thl.product import Product from generalresearch.models.thl.user import User @@ -11,13 +15,13 @@ class TestContest: """ @pytest.fixture(scope="function") - def user_1(self, user_factory, product) -> User: + def user_1(self, user_factory: Callable[..., User], product: Product) -> User: return user_factory(product=product) @pytest.fixture(scope="function") - def user_2(self, user_factory, product) -> User: + def user_2(self, user_factory: Callable[..., User], product: Product) -> User: return user_factory(product=product) @pytest.fixture(scope="function") - def user_3(self, user_factory, product) -> User: + def user_3(self, user_factory: Callable[..., User], product: Product) -> User: return user_factory(product=product) diff --git a/tests/models/thl/test_contest/test_leaderboard_contest.py b/tests/models/thl/test_contest/test_leaderboard_contest.py index 98f3215..8b714ee 100644 --- a/tests/models/thl/test_contest/test_leaderboard_contest.py +++ b/tests/models/thl/test_contest/test_leaderboard_contest.py @@ -7,8 +7,8 @@ from generalresearch.currency import USDCent from generalresearch.managers.leaderboard.manager import LeaderboardManager from generalresearch.models.thl.contest import ContestPrize from generalresearch.models.thl.contest.definitions import ( - ContestType, ContestPrizeKind, + ContestType, ) from generalresearch.models.thl.contest.leaderboard import ( LeaderboardContest, @@ -17,6 +17,7 @@ from generalresearch.models.thl.contest.utils import ( distribute_leaderboard_prizes, ) from generalresearch.models.thl.leaderboard import LeaderboardRow +from generalresearch.models.thl.product import Product from tests.models.thl.test_contest.test_contest import TestContest @@ -24,7 +25,7 @@ class TestLeaderboardContest(TestContest): @pytest.fixture def leaderboard_contest( - self, product, thl_redis, user_manager + self, product: Product, thl_redis, user_manager ) -> "LeaderboardContest": board_key = f"leaderboard:{product.uuid}:us:weekly:2025-05-26:complete_count" |
