Skip to content

Commit 2797ead

Browse files
committed
move the load test suite to new path
Signed-off-by: Yuchen Zhang <[email protected]>
1 parent 186ffbc commit 2797ead

File tree

7 files changed

+24
-24
lines changed

7 files changed

+24
-24
lines changed

src/nat/front_ends/mcp/load_test_utils/README.md renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/README.md

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@ uv pip install psutil
3737
Run a load test from the project root:
3838

3939
```bash
40-
python src/nat/front_ends/mcp/load_test_utils/cli.py \
41-
--config_file=src/nat/front_ends/mcp/load_test_utils/configs/config.yml
40+
python packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py \
41+
--config_file=packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/configs/config.yml
4242
```
4343

4444
Get help:
4545

4646
```bash
47-
python src/nat/front_ends/mcp/load_test_utils/cli.py --help
47+
python packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py --help
4848
```
4949

5050
## Configuration
@@ -147,35 +147,35 @@ Run load tests from the project root using the command-line interface:
147147

148148
```bash
149149
# Basic usage
150-
python src/nat/front_ends/mcp/load_test_utils/cli.py \
151-
--config_file=src/nat/front_ends/mcp/load_test_utils/configs/config.yml
150+
python packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py \
151+
--config_file=packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/configs/config.yml
152152
153153
# With verbose logging
154-
python src/nat/front_ends/mcp/load_test_utils/cli.py \
155-
--config_file=src/nat/front_ends/mcp/load_test_utils/configs/config.yml \
154+
python packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py \
155+
--config_file=packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/configs/config.yml \
156156
--verbose
157157
158158
# Short form
159-
python src/nat/front_ends/mcp/load_test_utils/cli.py \
160-
-c src/nat/front_ends/mcp/load_test_utils/configs/config.yml
159+
python packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py \
160+
-c packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/configs/config.yml
161161
```
162162

163163
### Python API
164164

165165
#### Using YAML Configuration
166166

167167
```python
168-
from nat.front_ends.mcp.load_test_utils import run_load_test_from_yaml
168+
from nat.test.mcp.load_test_utils import run_load_test_from_yaml
169169
170170
results = run_load_test_from_yaml(
171-
"src/nat/front_ends/mcp/load_test_utils/configs/config.yml"
171+
"packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/configs/config.yml"
172172
)
173173
```
174174

175175
#### Programmatic Usage
176176

177177
```python
178-
from nat.front_ends.mcp.load_test_utils import run_load_test
178+
from nat.test.mcp.load_test_utils import run_load_test
179179
180180
results = run_load_test(
181181
config_file="examples/getting_started/simple_calculator/configs/config.yml",

src/nat/front_ends/mcp/load_test_utils/__init__.py renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
This module provides utilities for load testing MCP servers.
1818
"""
1919

20-
from nat.front_ends.mcp.load_test_utils.load_tester import MCPLoadTest
21-
from nat.front_ends.mcp.load_test_utils.load_tester import run_load_test
22-
from nat.front_ends.mcp.load_test_utils.load_tester import run_load_test_from_yaml
20+
from nat.test.mcp.load_test_utils.load_tester import MCPLoadTest
21+
from nat.test.mcp.load_test_utils.load_tester import run_load_test
22+
from nat.test.mcp.load_test_utils.load_tester import run_load_test_from_yaml
2323

2424
__all__ = ["MCPLoadTest", "run_load_test", "run_load_test_from_yaml"]

src/nat/front_ends/mcp/load_test_utils/cli.py renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
import sys
2424
from pathlib import Path
2525

26-
from nat.front_ends.mcp.load_test_utils import run_load_test_from_yaml
26+
from nat.test.mcp.load_test_utils import run_load_test_from_yaml
2727

2828
# Configure logging
2929
logging.basicConfig(

src/nat/front_ends/mcp/load_test_utils/config_loader.py renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/config_loader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
1919

2020
import yaml
2121

22-
from nat.front_ends.mcp.load_test_utils.load_tester import LoadTestConfig
23-
from nat.front_ends.mcp.load_test_utils.load_tester import ToolCallConfig
22+
from nat.test.mcp.load_test_utils.load_tester import LoadTestConfig
23+
from nat.test.mcp.load_test_utils.load_tester import ToolCallConfig
2424

2525
logger = logging.getLogger(__name__)
2626

src/nat/front_ends/mcp/load_test_utils/load_tester.py renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/load_tester.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from mcp.client.streamable_http import streamablehttp_client
3131
from mcp.types import TextContent
3232

33-
from nat.front_ends.mcp.load_test_utils.report_generator import generate_summary_report
33+
from nat.test.mcp.load_test_utils.report_generator import generate_summary_report
3434

3535
logger = logging.getLogger(__name__)
3636

@@ -553,8 +553,8 @@ def run_load_test_from_yaml(yaml_config_path: str) -> dict[str, Any]:
553553
Returns:
554554
Dictionary containing test results and statistics
555555
"""
556-
from nat.front_ends.mcp.load_test_utils.config_loader import load_config_from_yaml
557-
from nat.front_ends.mcp.load_test_utils.config_loader import validate_config
556+
from nat.test.mcp.load_test_utils.config_loader import load_config_from_yaml
557+
from nat.test.mcp.load_test_utils.config_loader import validate_config
558558

559559
config = load_config_from_yaml(yaml_config_path)
560560
validate_config(config)

src/nat/front_ends/mcp/load_test_utils/report_generator.py renamed to packages/nvidia_nat_test/src/nat/test/mcp/load_test_utils/report_generator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@
1919
from typing import Any
2020

2121
if TYPE_CHECKING:
22-
from nat.front_ends.mcp.load_test_utils.load_tester import LoadTestConfig
23-
from nat.front_ends.mcp.load_test_utils.load_tester import MemorySample
24-
from nat.front_ends.mcp.load_test_utils.load_tester import ToolCallResult
22+
from nat.test.mcp.load_test_utils.load_tester import LoadTestConfig
23+
from nat.test.mcp.load_test_utils.load_tester import MemorySample
24+
from nat.test.mcp.load_test_utils.load_tester import ToolCallResult
2525

2626

2727
def generate_summary_report(

0 commit comments

Comments
 (0)