style: 优化代码质量和格式

Author: abc-git

examples/advanced_usage.py

@@ -21,9 +21,15 @@ def get_raw_data(data_source, size):
 def process_data(raw_data, algorithm):
     """处理数据生成器，依赖于raw_data"""
     if algorithm == "algo1":
-        return [{"id": item["id"], "processed": item["id"] * 2} for item in raw_data]
+        return [
+            {"id": item["id"], "processed": item["id"] * 2}
+            for item in raw_data
+        ]
     else:
-        return [{"id": item["id"], "processed": item["id"] * 3} for item in raw_data]
+        return [
+            {"id": item["id"], "processed": item["id"] * 3}
+            for item in raw_data
+        ]
 
 
 @param_generator
@@ -43,18 +49,20 @@ def validate_results(processed_data, threshold):
 @pytest.mark.parametrize("algorithm", ["algo1", "algo2"])
 @pytest.mark.parametrize("threshold", [0.5, 8.0])
 def test_dynamic_params_nesting(
-    data_source, size, algorithm, threshold,
-    raw_data, processed_data, is_valid
+    data_source, size, algorithm, threshold, raw_data, processed_data, is_valid
 ):
     """多个动态参数嵌套依赖测试示例"""
     # 验证原始数据
     assert len(raw_data) == size
     assert all(item["source"] == data_source for item in raw_data)
-    
+
     # 验证处理后的数据
     assert len(processed_data) == size
-    assert all(item["id"] == raw_item["id"] for item, raw_item in zip(processed_data, raw_data))
-    
+    assert all(
+        item["id"] == raw_item["id"]
+        for item, raw_item in zip(processed_data, raw_data)
+    )
+
     # 验证结果
     assert isinstance(is_valid, bool)
 
@@ -86,15 +94,17 @@ def session_scoped_data():
 
 class TestScopesUsage:
     """作用域使用测试类示例"""
-    
+
     @with_dynamic_params(
         func_data=function_scoped_data,
         class_data=class_scoped_data,
         mod_data=module_scoped_data,
-        session_data=session_scoped_data
+        session_data=session_scoped_data,
     )
     @pytest.mark.parametrize("input_value", [1, 2])
-    def test_scope_management(self, input_value, func_data, class_data, mod_data, session_data):
+    def test_scope_management(
+        self, input_value, func_data, class_data, mod_data, session_data
+    ):
         """作用域管理测试示例"""
         assert func_data == f"func_{input_value}"
         assert class_data == "class_shared"
@@ -118,16 +128,13 @@ def uncached_data():
     return time.time()
 
 
-@with_dynamic_params(
-    cached_result=cached_data,
-    uncached_result=uncached_data
-)
+@with_dynamic_params(cached_result=cached_data, uncached_result=uncached_data)
 @pytest.mark.parametrize("input_value", [1, 2])
 def test_cache_control(input_value, cached_result, uncached_result):
     """缓存控制测试示例"""
     # 验证缓存结果
     assert cached_result == input_value * 10
-    
+
     # 验证非缓存结果（每次都不同）
     assert isinstance(uncached_result, float)
 
@@ -149,16 +156,13 @@ def eager_data(input_value):
     return input_value * 3
 
 
-@with_dynamic_params(
-    lazy_result=lazy_data,
-    eager_result=eager_data
-)
+@with_dynamic_params(lazy_result=lazy_data, eager_result=eager_data)
 @pytest.mark.parametrize("input_value", [1, 2])
 def test_lazy_loading(input_value, lazy_result, eager_result):
     """懒加载控制测试示例"""
     # 验证懒加载结果
     assert lazy_result == input_value * 5
-    
+
     # 验证非懒加载结果
     assert eager_result == input_value * 3
 

examples/basic_usage.py

@@ -57,7 +57,7 @@ def generate_config(environment, feature_flag):
     return {
         "env": environment["env"],
         "feature": feature_flag,
-        "timeout": environment["timeout"] + 10
+        "timeout": environment["timeout"] + 10,
     }
 
 
@@ -92,11 +92,7 @@ def app_config(database_config):
 @param_generator
 def generate_test_data(app_config, test_type):
     """依赖于嵌套fixture的参数生成器"""
-    return {
-        "config": app_config,
-        "type": test_type,
-        "data": [1, 2, 3]
-    }
+    return {"config": app_config, "type": test_type, "data": [1, 2, 3]}
 
 
 @with_dynamic_params(test_data=generate_test_data)

setup.py

@@ -6,14 +6,10 @@
     packages=find_packages(where="src"),  # 只查找src目录下的包
     package_dir={"": "src"},
     install_requires=["pytest"],
-    entry_points={
-        "pytest11": [
-            "dynamic-params = dynamic_params.plugin"
-        ]
-    },
+    entry_points={"pytest11": ["dynamic-params = dynamic_params.plugin"]},
     classifiers=[
         "Framework :: Pytest",
         "Programming Language :: Python :: 3",
         "Operating System :: OS Independent",
     ],
-)
+)

src/__init__.py

@@ -5,20 +5,33 @@
 allowing parameters to be computed based on other parameters or fixtures.
 """
 
-from .dynamic_params import *
+from .dynamic_params import (
+    ParamGenerator,
+    LazyResult,
+    GeneratorRegistry,
+    DynamicParamError,
+    MissingParameterError,
+    InvalidGeneratorError,
+    param_generator,
+    with_dynamic_params,
+    pytest_configure,
+    pytest_generate_tests,
+    DynamicParamConfig,
+)
 
 __version__ = "0.1.0"
-__author__ = "pytest-dynamic-params"
+__author__ = "Your Name"
+
 __all__ = [
-    'ParamGenerator',
-    'LazyResult', 
-    'GeneratorRegistry',
-    'DynamicParamError',
-    'MissingParameterError',
-    'InvalidGeneratorError',
-    'param_generator',
-    'with_dynamic_params',
-    'pytest_configure',
-    'pytest_generate_tests',
-    'DynamicParamConfig'
-]
+    "ParamGenerator",
+    "LazyResult",
+    "GeneratorRegistry",
+    "DynamicParamError",
+    "MissingParameterError",
+    "InvalidGeneratorError",
+    "param_generator",
+    "with_dynamic_params",
+    "pytest_configure",
+    "pytest_generate_tests",
+    "DynamicParamConfig",
+]

src/dynamic_params/__init__.py

@@ -2,20 +2,24 @@
 from .core.generator import ParamGenerator
 from .core.registry import GeneratorRegistry
 from .decorators import param_generator, with_dynamic_params
-from .errors import DynamicParamError, InvalidGeneratorError, MissingParameterError
+from .errors import (
+    DynamicParamError,
+    InvalidGeneratorError,
+    MissingParameterError
+)
 from .lazy import LazyResult
 from .plugin import pytest_configure, pytest_generate_tests
 
 __all__ = [
-    'ParamGenerator',
-    'LazyResult',
-    'GeneratorRegistry',
-    'DynamicParamError',
-    'MissingParameterError',
-    'InvalidGeneratorError',
-    'param_generator',
-    'with_dynamic_params',
-    'pytest_configure',
-    'pytest_generate_tests',
-    'DynamicParamConfig'
-]
+    "ParamGenerator",
+    "LazyResult",
+    "GeneratorRegistry",
+    "DynamicParamError",
+    "MissingParameterError",
+    "InvalidGeneratorError",
+    "param_generator",
+    "with_dynamic_params",
+    "pytest_configure",
+    "pytest_generate_tests",
+    "DynamicParamConfig",
+]