You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.8 KiB
81 lines
2.8 KiB
import pytest
|
|
import json
|
|
|
|
from swarms.utils.str_to_dict import str_to_dict
|
|
|
|
|
|
class TestStrToDict:
|
|
"""Test cases for the str_to_dict function."""
|
|
|
|
def test_valid_json_string(self):
|
|
"""Test converting a valid JSON string to dictionary."""
|
|
result = str_to_dict('{"key": "value"}')
|
|
assert result == {"key": "value"}
|
|
|
|
def test_nested_json_string(self):
|
|
"""Test converting a nested JSON string."""
|
|
result = str_to_dict('{"a": 1, "b": {"c": 2}}')
|
|
assert result == {"a": 1, "b": {"c": 2}}
|
|
|
|
def test_list_in_json_string(self):
|
|
"""Test converting JSON string containing a list."""
|
|
result = str_to_dict('{"items": [1, 2, 3]}')
|
|
assert result == {"items": [1, 2, 3]}
|
|
|
|
def test_empty_json_object(self):
|
|
"""Test converting an empty JSON object."""
|
|
result = str_to_dict("{}")
|
|
assert result == {}
|
|
|
|
def test_json_with_numbers(self):
|
|
"""Test converting JSON string with various number types."""
|
|
result = str_to_dict('{"int": 42, "float": 3.14, "negative": -5}')
|
|
assert result == {"int": 42, "float": 3.14, "negative": -5}
|
|
|
|
def test_json_with_booleans(self):
|
|
"""Test converting JSON string with boolean values."""
|
|
result = str_to_dict('{"true_val": true, "false_val": false}')
|
|
assert result == {"true_val": True, "false_val": False}
|
|
|
|
def test_json_with_null(self):
|
|
"""Test converting JSON string with null value."""
|
|
result = str_to_dict('{"value": null}')
|
|
assert result == {"value": None}
|
|
|
|
def test_invalid_json_raises_error(self):
|
|
"""Test that invalid JSON raises JSONDecodeError."""
|
|
with pytest.raises(json.JSONDecodeError):
|
|
str_to_dict('{"invalid": json}') # Invalid JSON
|
|
|
|
def test_complex_nested_structure(self):
|
|
"""Test converting a complex nested JSON structure."""
|
|
json_str = '''
|
|
{
|
|
"user": {
|
|
"name": "John",
|
|
"age": 30,
|
|
"active": true
|
|
},
|
|
"tags": ["python", "testing"],
|
|
"metadata": null
|
|
}
|
|
'''
|
|
result = str_to_dict(json_str)
|
|
assert result["user"]["name"] == "John"
|
|
assert result["user"]["age"] == 30
|
|
assert result["tags"] == ["python", "testing"]
|
|
assert result["metadata"] is None
|
|
|
|
def test_retries_parameter(self):
|
|
"""Test that retries parameter works correctly."""
|
|
# This should succeed on first try
|
|
result = str_to_dict('{"test": 1}', retries=1)
|
|
assert result == {"test": 1}
|
|
|
|
def test_json_with_unicode_characters(self):
|
|
"""Test converting JSON string with unicode characters."""
|
|
result = str_to_dict('{"emoji": "🐍", "text": "你好"}')
|
|
assert result["emoji"] == "🐍"
|
|
assert result["text"] == "你好"
|
|
|