Skip to content

Commit c5bc7d3

Browse files
committed
remove verbose request options in unit test parameters
1 parent 74ca76c commit c5bc7d3

2 files changed

Lines changed: 3 additions & 43 deletions

File tree

tests/sdk/test_async_scorer.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,7 @@ async def test_get_info(self, mock_async_client: AsyncMock, scorer_view: MockSco
3535
mock_async_client.scenarios.scorers.retrieve = AsyncMock(return_value=scorer_view)
3636

3737
scorer = AsyncScorer(mock_async_client, "scorer_123")
38-
result = await scorer.get_info(
39-
extra_headers={"X-Custom": "value"},
40-
extra_query={"param": "value"},
41-
extra_body={"key": "value"},
42-
timeout=30.0,
43-
)
38+
result = await scorer.get_info()
4439

4540
assert result == scorer_view
4641
mock_async_client.scenarios.scorers.retrieve.assert_called_once()
@@ -55,10 +50,6 @@ async def test_update(self, mock_async_client: AsyncMock) -> None:
5550
result = await scorer.update(
5651
type="updated_scorer",
5752
bash_script="echo 'score=1.0'",
58-
extra_headers={"X-Custom": "value"},
59-
extra_query={"param": "value"},
60-
extra_body={"key": "value"},
61-
timeout=30.0,
6253
)
6354

6455
assert result == update_response
@@ -77,10 +68,6 @@ async def test_validate(self, mock_async_client: AsyncMock) -> None:
7768
scorer = AsyncScorer(mock_async_client, "scorer_123")
7869
result = await scorer.validate(
7970
scoring_context={"test": "context"},
80-
extra_headers={"X-Custom": "value"},
81-
extra_query={"param": "value"},
82-
extra_body={"key": "value"},
83-
timeout=30.0,
8471
)
8572

8673
assert result == validate_response

tests/sdk/test_scorer.py

Lines changed: 2 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -32,21 +32,10 @@ def test_get_info(self, mock_client: Mock, scorer_view: MockScorerView) -> None:
3232
mock_client.scenarios.scorers.retrieve.return_value = scorer_view
3333

3434
scorer = Scorer(mock_client, "scorer_123")
35-
result = scorer.get_info(
36-
extra_headers={"X-Custom": "value"},
37-
extra_query={"param": "value"},
38-
extra_body={"key": "value"},
39-
timeout=30.0,
40-
)
35+
result = scorer.get_info()
4136

4237
assert result == scorer_view
43-
mock_client.scenarios.scorers.retrieve.assert_called_once_with(
44-
"scorer_123",
45-
extra_headers={"X-Custom": "value"},
46-
extra_query={"param": "value"},
47-
extra_body={"key": "value"},
48-
timeout=30.0,
49-
)
38+
mock_client.scenarios.scorers.retrieve.assert_called_once_with("scorer_123")
5039

5140
def test_update(self, mock_client: Mock) -> None:
5241
"""Test update method."""
@@ -57,21 +46,13 @@ def test_update(self, mock_client: Mock) -> None:
5746
result = scorer.update(
5847
type="updated_scorer",
5948
bash_script="echo 'score=1.0'",
60-
extra_headers={"X-Custom": "value"},
61-
extra_query={"param": "value"},
62-
extra_body={"key": "value"},
63-
timeout=30.0,
6449
)
6550

6651
assert result == update_response
6752
mock_client.scenarios.scorers.update.assert_called_once_with(
6853
"scorer_123",
6954
type="updated_scorer",
7055
bash_script="echo 'score=1.0'",
71-
extra_headers={"X-Custom": "value"},
72-
extra_query={"param": "value"},
73-
extra_body={"key": "value"},
74-
timeout=30.0,
7556
)
7657

7758
def test_validate(self, mock_client: Mock) -> None:
@@ -86,18 +67,10 @@ def test_validate(self, mock_client: Mock) -> None:
8667
scorer = Scorer(mock_client, "scorer_123")
8768
result = scorer.validate(
8869
scoring_context={"test": "context"},
89-
extra_headers={"X-Custom": "value"},
90-
extra_query={"param": "value"},
91-
extra_body={"key": "value"},
92-
timeout=30.0,
9370
)
9471

9572
assert result == validate_response
9673
mock_client.scenarios.scorers.validate.assert_called_once_with(
9774
"scorer_123",
9875
scoring_context={"test": "context"},
99-
extra_headers={"X-Custom": "value"},
100-
extra_query={"param": "value"},
101-
extra_body={"key": "value"},
102-
timeout=30.0,
10376
)

0 commit comments

Comments
 (0)