def test_register(add_type): YamlParser.register() assert add_type.mock_calls == [ call('application/vnd.yaml', '.yaml'), call('application/vnd.yaml', '.yml'), call('application/x-yaml', '.yaml'), call('application/x-yaml', '.yml'), call('application/yaml', '.yaml'), call('application/yaml', '.yml'), call('text/vnd.yaml', '.yaml'), call('text/vnd.yaml', '.yml'), call('text/x-yaml', '.yaml'), call('text/x-yaml', '.yml'), call('text/yaml', '.yaml'), call('text/yaml', '.yml'), ]
def test_loads(): parser = YamlParser() result = parser.loads("[1, 2, 3]") assert result == [1, 2, 3]
def test_load(): parser = YamlParser() result = parser.load(StringIO("[1, 2, 3]")) assert result == [1, 2, 3]