@@ -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