TEST: Rename converter/ to prompt_converter/ and add 19 converter tests (#1594)

Author: romanlutz

…nverter/test_add_image_text_converter.py …nverter/test_add_image_text_converter.pytests/unit/converter/test_add_image_text_converter.py renamed to tests/unit/prompt_converter/test_add_image_text_converter.py tests/unit/converter/test_add_image_text_converter.py renamed to tests/unit/prompt_converter/test_add_image_text_converter.py

@@ -0,0 +1,53 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+import pytest
+
+pytest.importorskip("art")
+
+from pyrit.prompt_converter import AsciiArtConverter, ConverterResult
+
+
+@pytest.mark.asyncio
+async def test_ascii_art_converter_basic():
+    converter = AsciiArtConverter(font="block")
+    result = await converter.convert_async(prompt="hi", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_type == "text"
+    assert len(result.output_text) > 0
+    assert "\n" in result.output_text
+
+
+@pytest.mark.asyncio
+async def test_ascii_art_converter_default_random_font():
+    converter = AsciiArtConverter()
+    result = await converter.convert_async(prompt="test", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert len(result.output_text) > 0
+
+
+@pytest.mark.asyncio
+async def test_ascii_art_converter_empty():
+    converter = AsciiArtConverter(font="block")
+    result = await converter.convert_async(prompt="", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_ascii_art_converter_input_not_supported():
+    converter = AsciiArtConverter()
+    with pytest.raises(ValueError, match="Input type not supported"):
+        await converter.convert_async(prompt="test", input_type="image_path")
+
+
+def test_ascii_art_converter_input_supported():
+    converter = AsciiArtConverter()
+    assert converter.input_supported("text") is True
+    assert converter.input_supported("image_path") is False
+
+
+def test_ascii_art_converter_output_supported():
+    converter = AsciiArtConverter()
+    assert converter.output_supported("text") is True
+    assert converter.output_supported("image_path") is False

…verter/test_add_image_video_converter.py …verter/test_add_image_video_converter.pytests/unit/converter/test_add_image_video_converter.py renamed to tests/unit/prompt_converter/test_add_image_video_converter.py tests/unit/converter/test_add_image_video_converter.py renamed to tests/unit/prompt_converter/test_add_image_video_converter.py

@@ -0,0 +1,53 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+import pytest
+
+from pyrit.prompt_converter import AsciiSmugglerConverter, ConverterResult
+
+
+@pytest.mark.asyncio
+async def test_ascii_smuggler_encode_basic():
+    converter = AsciiSmugglerConverter(action="encode")
+    result = await converter.convert_async(prompt="hi", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_type == "text"
+    for char in result.output_text:
+        assert ord(char) > 0xE0000
+
+
+@pytest.mark.asyncio
+async def test_ascii_smuggler_decode_roundtrip():
+    encoder = AsciiSmugglerConverter(action="encode")
+    encoded = await encoder.convert_async(prompt="hello", input_type="text")
+
+    decoder = AsciiSmugglerConverter(action="decode")
+    decoded = await decoder.convert_async(prompt=encoded.output_text, input_type="text")
+    assert decoded.output_text == "hello"
+
+
+@pytest.mark.asyncio
+async def test_ascii_smuggler_with_unicode_tags():
+    converter = AsciiSmugglerConverter(action="encode", unicode_tags=True)
+    result = await converter.convert_async(prompt="hi", input_type="text")
+    assert result.output_text.startswith(chr(0xE0001))
+    assert result.output_text.endswith(chr(0xE007F))
+
+
+@pytest.mark.asyncio
+async def test_ascii_smuggler_empty():
+    converter = AsciiSmugglerConverter(action="encode")
+    result = await converter.convert_async(prompt="", input_type="text")
+    assert result.output_text == ""
+
+
+def test_ascii_smuggler_invalid_action():
+    with pytest.raises(ValueError):
+        AsciiSmugglerConverter(action="invalid")
+
+
+@pytest.mark.asyncio
+async def test_ascii_smuggler_input_not_supported():
+    converter = AsciiSmugglerConverter(action="encode")
+    with pytest.raises(ValueError, match="Input type not supported"):
+        await converter.convert_async(prompt="test", input_type="image_path")

…nverter/test_add_text_image_converter.py …nverter/test_add_text_image_converter.pytests/unit/converter/test_add_text_image_converter.py renamed to tests/unit/prompt_converter/test_add_text_image_converter.py tests/unit/converter/test_add_text_image_converter.py renamed to tests/unit/prompt_converter/test_add_text_image_converter.py

@@ -0,0 +1,80 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+
+import pytest
+
+from pyrit.prompt_converter import AtbashConverter, ConverterResult
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_basic():
+    converter = AtbashConverter()
+    result = await converter.convert_async(prompt="abc", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_text == "zyx"
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_uppercase():
+    converter = AtbashConverter()
+    result = await converter.convert_async(prompt="ABC", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_text == "ZYX"
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_mixed():
+    converter = AtbashConverter()
+    result = await converter.convert_async(prompt="Hello", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_text == "Svool"
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_with_description():
+    converter = AtbashConverter(append_description=True)
+    result = await converter.convert_async(prompt="hello", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_type == "text"
+    # The encoded prompt should be present in the output
+    assert "svool" in result.output_text
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_empty_string():
+    converter = AtbashConverter()
+    result = await converter.convert_async(prompt="", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_text == ""
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_numbers():
+    converter = AtbashConverter()
+    result = await converter.convert_async(prompt="012", input_type="text")
+    assert isinstance(result, ConverterResult)
+    assert result.output_text == "987"
+    assert result.output_type == "text"
+
+
+@pytest.mark.asyncio
+async def test_atbash_converter_input_not_supported():
+    converter = AtbashConverter()
+    with pytest.raises(ValueError):
+        await converter.convert_async(prompt="hello", input_type="image_path")
+
+
+def test_atbash_converter_input_supported():
+    converter = AtbashConverter()
+    assert converter.input_supported("text") is True
+    assert converter.input_supported("image_path") is False
+
+
+def test_atbash_converter_output_supported():
+    converter = AtbashConverter()
+    assert converter.output_supported("text") is True
+    assert converter.output_supported("image_path") is False