Skip to content

Commit 5b2920e

Browse files
authored
use yaml.safe_load in tests (#1685)
1 parent 448c7b7 commit 5b2920e

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/json_body_serializer.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@
66
import yaml
77

88
if TYPE_CHECKING:
9-
from yaml import Dumper, Loader
9+
from yaml import Dumper
1010
else:
1111
try:
12-
from yaml import CDumper as Dumper, CLoader as Loader
13-
except ImportError:
14-
from yaml import Dumper, Loader
12+
from yaml import CDumper as Dumper
13+
except ImportError: # pragma: no cover
14+
from yaml import Dumper
1515

1616
FILTERED_HEADER_PREFIXES = ['anthropic-', 'cf-', 'x-']
1717
FILTERED_HEADERS = {'authorization', 'date', 'request-id', 'server', 'user-agent', 'via', 'set-cookie', 'api-key'}
@@ -35,7 +35,7 @@ def str_presenter(dumper: Dumper, data: str):
3535

3636

3737
def deserialize(cassette_string: str):
38-
cassette_dict = yaml.load(cassette_string, Loader=Loader)
38+
cassette_dict = yaml.safe_load(cassette_string)
3939
for interaction in cassette_dict['interactions']:
4040
for kind, data in interaction.items():
4141
parsed_body = data.pop('parsed_body', None)
@@ -45,7 +45,7 @@ def deserialize(cassette_string: str):
4545
return cassette_dict
4646

4747

48-
def serialize(cassette_dict: Any):
48+
def serialize(cassette_dict: Any): # pragma: lax no cover
4949
for interaction in cassette_dict['interactions']:
5050
for _kind, data in interaction.items():
5151
headers: dict[str, list[str]] = data.get('headers', {})

0 commit comments

Comments
 (0)