From 4a6fb048ca05747fb9f2c2d2a37c1ebd5671216d Mon Sep 17 00:00:00 2001 From: ed cuss Date: Mon, 29 Sep 2025 19:44:32 +0100 Subject: [PATCH] fix: ruff fixes for test case --- mock_data/mock_case_9/src/case_9/mod_generics.py | 7 ++----- mock_data/mock_case_9/src/case_9/mod_utils.py | 2 +- tests/test_main.py | 10 +++------- 3 files changed, 6 insertions(+), 13 deletions(-) diff --git a/mock_data/mock_case_9/src/case_9/mod_generics.py b/mock_data/mock_case_9/src/case_9/mod_generics.py index 50bd5ac..fe02cf0 100644 --- a/mock_data/mock_case_9/src/case_9/mod_generics.py +++ b/mock_data/mock_case_9/src/case_9/mod_generics.py @@ -1,4 +1,4 @@ -from typing import Optional +from __future__ import annotations from case_9.mod_utils import process_data @@ -8,10 +8,7 @@ def process_list(items: list[str]) -> list[str]: def create_mapping(keys: list[str], values: list[int]) -> dict[str, int]: - result: dict[str, int] = {} - for key, value in zip(keys, values): - result[key] = value - return result + return dict(zip(keys, values, strict=False)) def find_item(items: list[str], target: str) -> str | None: diff --git a/mock_data/mock_case_9/src/case_9/mod_utils.py b/mock_data/mock_case_9/src/case_9/mod_utils.py index 01b2067..6a208d9 100644 --- a/mock_data/mock_case_9/src/case_9/mod_utils.py +++ b/mock_data/mock_case_9/src/case_9/mod_utils.py @@ -1,6 +1,6 @@ from typing import TypeVar -T = TypeVar('T') +T = TypeVar("T") def process_data[T](data: list[T]) -> list[T]: diff --git a/tests/test_main.py b/tests/test_main.py index 16bac90..164a814 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -218,7 +218,7 @@ def test_main(src_root): "mock_data/mock_case_9/src/case_9", { "result/mock_data/mock_case_9/src/case_9/__init__.py": "", - "result/mock_data/mock_case_9/src/case_9/mod_generics.py": "from __future__ import annotations\n\nfrom typing import Optional, TypeVar\n" + "result/mock_data/mock_case_9/src/case_9/mod_generics.py": "from __future__ import annotations\n\nfrom typing import TypeVar\n" "\n" "from case_9.mod_utils import T\n" "\n" @@ -228,16 +228,12 @@ def test_main(src_root): "\n" "\n" "def create_mapping(keys: list[str], values: list[int]) -> dict[str, int]:\n" - " result: dict[str, int] = {}\n" - " for key, value in zip(keys, values):\n" - " result[key] = value\n" - " return result\n" + " return dict(zip(keys, values, strict=False))\n" "\n" "\n" "def process_data[T](data: list[T]) -> list[T]:\n" " return [item for item in data if item is not None]\n", - "result/mock_data/mock_case_9/src/case_9/mod_generics_find_item.py": "from __future__ import annotations\n\nfrom typing import Optional\n" - "\n" + "result/mock_data/mock_case_9/src/case_9/mod_generics_find_item.py": "from __future__ import annotations\n\n" "\n" "def find_item(items: list[str], target: str) -> str | None:\n" " for item in items:\n"