|
6 | 6 | """ |
7 | 7 |
|
8 | 8 | from dataclasses import dataclass |
9 | | -from typing import Dict, List, Optional |
10 | 9 | from datetime import datetime |
11 | 10 | from enum import Enum |
12 | 11 |
|
@@ -34,11 +33,11 @@ class ConnectionConfig: |
34 | 33 | host: str |
35 | 34 | port: int |
36 | 35 | database: str |
37 | | - username: Optional[str] = None |
38 | | - password: Optional[str] = None |
| 36 | + username: str | None = None |
| 37 | + password: str | None = None |
39 | 38 | connection_timeout: float = 10.0 |
40 | 39 |
|
41 | | - def validate(self) -> List[str]: |
| 40 | + def validate(self) -> list[str]: |
42 | 41 | """ |
43 | 42 | Validate configuration parameters. |
44 | 43 |
|
@@ -69,9 +68,9 @@ class BenchmarkConfiguration: |
69 | 68 | concurrent_connections: int = 1 |
70 | 69 | warmup_queries: int = 100 |
71 | 70 | random_seed: int = 42 |
72 | | - connection_configs: Optional[Dict[str, ConnectionConfig]] = None |
| 71 | + connection_configs: dict[str, ConnectionConfig] | None = None |
73 | 72 |
|
74 | | - def validate(self) -> List[str]: |
| 73 | + def validate(self) -> list[str]: |
75 | 74 | """ |
76 | 75 | Validate configuration per functional requirements. |
77 | 76 |
|
@@ -120,11 +119,11 @@ class PerformanceResult: |
120 | 119 | timestamp: datetime |
121 | 120 | elapsed_ms: float |
122 | 121 | success: bool |
123 | | - error_message: Optional[str] = None |
| 122 | + error_message: str | None = None |
124 | 123 | row_count: int = 0 |
125 | | - resource_usage: Optional[Dict] = None |
| 124 | + resource_usage: dict | None = None |
126 | 125 |
|
127 | | - def validate(self) -> List[str]: |
| 126 | + def validate(self) -> list[str]: |
128 | 127 | """ |
129 | 128 | Validate result integrity. |
130 | 129 |
|
@@ -165,9 +164,9 @@ class MethodResults: |
165 | 164 | latency_p50_ms: float |
166 | 165 | latency_p95_ms: float |
167 | 166 | latency_p99_ms: float |
168 | | - by_category: Dict[str, CategoryMetrics] |
| 167 | + by_category: dict[str, CategoryMetrics] |
169 | 168 |
|
170 | | - def validate(self) -> List[str]: |
| 169 | + def validate(self) -> list[str]: |
171 | 170 | """ |
172 | 171 | Validate aggregated results. |
173 | 172 |
|
@@ -205,12 +204,12 @@ class BenchmarkReport: |
205 | 204 | start_time: datetime |
206 | 205 | end_time: datetime |
207 | 206 | total_duration_seconds: float |
208 | | - method_results: Dict[str, MethodResults] |
209 | | - raw_results: List[PerformanceResult] |
210 | | - validation_errors: List[str] |
| 207 | + method_results: dict[str, MethodResults] |
| 208 | + raw_results: list[PerformanceResult] |
| 209 | + validation_errors: list[str] |
211 | 210 | state: BenchmarkState = BenchmarkState.INITIALIZING |
212 | 211 |
|
213 | | - def to_json(self) -> Dict: |
| 212 | + def to_json(self) -> dict: |
214 | 213 | """ |
215 | 214 | Export report as JSON (FR-010). |
216 | 215 |
|
@@ -239,7 +238,7 @@ def to_json(self) -> Dict: |
239 | 238 | } |
240 | 239 | } |
241 | 240 |
|
242 | | - def to_table_rows(self) -> List[List]: |
| 241 | + def to_table_rows(self) -> list[list]: |
243 | 242 | """ |
244 | 243 | Export report as table rows for console display (FR-010). |
245 | 244 |
|
|
0 commit comments