aboutsummaryrefslogtreecommitdiff
path: root/tests/models/thl/test_contest
diff options
context:
space:
mode:
authorMax Nanis2026-08-18 16:35:11 -0700
committerMax Nanis2026-08-18 16:35:11 -0700
commit0742210fc7eaa54d8068bfd18414158dc9b33672 (patch)
tree5f42b671ccbd9cc44f1f62229cac16a5e455317d /tests/models/thl/test_contest
parent6b4f8d6b8700a65bdf74beb3d95c969a81464a56 (diff)
downloadgeneralresearch-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.py10
-rw-r--r--tests/models/thl/test_contest/test_leaderboard_contest.py5
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"