23 lines
544 B
Python
23 lines
544 B
Python
|
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)
|