Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tests/unit/models/test_seed.py
Original file line number Diff line number Diff line change
Expand Up @@ -582,18 +582,18 @@ async def test_hashes_generated_files():
os.remove(filename)


async def test_memory_encoding_metadata_image(sqlite_instance):
async def test_memory_encoding_metadata_image(tmp_path, sqlite_instance):
mock_image = Image.new("RGB", (400, 300), (255, 255, 255))
mock_image.save("test.png")
image_path = str(tmp_path / "test.png")
mock_image.save(image_path)
sp = SeedPrompt(
value="test.png",
value=image_path,
data_type="image_path",
)
await sqlite_instance.add_seeds_to_memory_async(seeds=[sp], added_by="test")
entry = sqlite_instance.get_seeds()[0]
assert len(entry.metadata) == 1
assert entry.metadata["format"] == "png"
os.remove("test.png")


@patch("pyrit.models.seeds.seed_prompt.TinyTag")
Expand Down
8 changes: 4 additions & 4 deletions tests/unit/prompt_converter/test_add_text_image_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,18 +85,18 @@ async def test_add_text_image_converter_invalid_input_image() -> None:
assert await converter.convert_async(prompt="mock_image.png", input_type="image_path") # type: ignore[arg-type]


async def test_add_text_image_converter_convert_async(sqlite_instance) -> None:
async def test_add_text_image_converter_convert_async(tmp_path, sqlite_instance) -> None:
converter = AddTextImageConverter(text_to_add="test")
mock_image = Image.new("RGB", (400, 300), (255, 255, 255))
mock_image.save("test.png")
image_path = str(tmp_path / "test.png")
mock_image.save(image_path)

converted_image = await converter.convert_async(prompt="test.png", input_type="image_path")
converted_image = await converter.convert_async(prompt=image_path, input_type="image_path")
assert converted_image
assert converted_image.output_text
assert converted_image.output_type == "image_path"
assert os.path.exists(converted_image.output_text)
os.remove(converted_image.output_text)
os.remove("test.png")


def test_text_image_converter_input_supported():
Expand Down
6 changes: 2 additions & 4 deletions tests/unit/prompt_converter/test_qr_code_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
# Licensed under the MIT license.

import os
from pathlib import Path
from unittest.mock import patch

import pytest
Expand Down Expand Up @@ -50,17 +49,16 @@ async def test_qr_code_converter_invalid_prompt() -> None:
await converter.convert_async(prompt="", input_type="text")


async def test_qr_code_converter_convert_async() -> None:
async def test_qr_code_converter_convert_async(tmp_path) -> None:
converter = QRCodeConverter()
with patch.object(converter._img_serializer, "get_data_filename") as mock_get_data_filename:
expected_filename = Path("sample_file.png").resolve()
expected_filename = tmp_path / "sample_file.png"
mock_get_data_filename.return_value = expected_filename
qr = await converter.convert_async(prompt="Sample prompt", input_type="text")
assert qr
assert str(qr.output_text) == str(expected_filename)
assert qr.output_type == "image_path"
assert os.path.exists(qr.output_text)
os.remove(qr.output_text)


def test_text_image_converter_input_supported():
Expand Down
Loading