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.
46 lines
1.5 KiB
46 lines
1.5 KiB
import pytest
|
|
from swarms.utils.xml_utils import dict_to_xml, to_xml_string
|
|
import xml.etree.ElementTree as ET
|
|
|
|
def test_dict_to_xml_simple():
|
|
d = {"foo": "bar", "baz": 1}
|
|
elem = dict_to_xml("root", d)
|
|
xml_str = ET.tostring(elem, encoding="unicode")
|
|
assert "<foo>bar</foo>" in xml_str
|
|
assert "<baz>1</baz>" in xml_str
|
|
|
|
def test_dict_to_xml_nested():
|
|
d = {"foo": {"bar": "baz"}}
|
|
elem = dict_to_xml("root", d)
|
|
xml_str = ET.tostring(elem, encoding="unicode")
|
|
assert "<foo>" in xml_str and "<bar>baz</bar>" in xml_str
|
|
|
|
def test_dict_to_xml_list():
|
|
d = {"items": [1, 2, 3]}
|
|
elem = dict_to_xml("root", d)
|
|
xml_str = ET.tostring(elem, encoding="unicode")
|
|
assert xml_str.count("<item>") == 3
|
|
assert "<item>1</item>" in xml_str
|
|
|
|
def test_to_xml_string_dict():
|
|
d = {"foo": "bar"}
|
|
xml = to_xml_string(d, root_tag="root")
|
|
assert xml.startswith("<root>") and "<foo>bar</foo>" in xml
|
|
|
|
def test_to_xml_string_list():
|
|
data = [{"a": 1}, {"b": 2}]
|
|
xml = to_xml_string(data, root_tag="root")
|
|
assert xml.startswith("<root>") and xml.count("<item>") == 2
|
|
|
|
def test_to_xml_string_scalar():
|
|
xml = to_xml_string("hello", root_tag="root")
|
|
assert xml == "<root>hello</root>"
|
|
|
|
def test_dict_to_xml_edge_cases():
|
|
d = {"empty": [], "none": None, "bool": True}
|
|
elem = dict_to_xml("root", d)
|
|
xml_str = ET.tostring(elem, encoding="unicode")
|
|
assert "<empty />" in xml_str or "<empty></empty>" in xml_str
|
|
assert "<none>None</none>" in xml_str
|
|
assert "<bool>True</bool>" in xml_str
|