참고소스 수정본

참고/guardrails-main/tests/unit_tests/utils/test_xml_utils.py

@@ -0,0 +1,22 @@
+import pytest
+
+from guardrails.utils.xml_utils import cast_xml_to_string
+
+
+@pytest.mark.parametrize(
+    "xml_input, expected_output",
+    [
+        # str
+        ("hello", "hello"),
+        # bytes
+        (b"hello", "hello"),
+        # bytearray
+        (bytearray(b"hello"), "hello"),
+        # memoryview
+        (memoryview(b"hello"), "hello"),
+    ],
+)
+def test_xml_element_cast(xml_input, expected_output):
+    cast_output = cast_xml_to_string(xml_input)
+    assert cast_output == expected_output
+    assert isinstance(cast_output, str)