mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-11-10 22:47:02 +01:00
Krzysztof kuhy Rudnicki
b60cb4972a
android_ui, app_icons, token_audit, focus_policy and random_jpg are now entirely under 250 lines. Only wsg_grabber's six test modules remain in python_pkg. android_ui/driver.py 411 -> 205 took two cuts and a class split: the element and error types plus the XML parser to _elements.py, and the text-entry methods to a TextEntryMixin, using the ABC-with-abstract- methods shape because the tests call them as bound attributes. app_icons/glyphs.py 291 -> 93 by moving the twelve path fragments out. Three intermediate failures, all module-level globals that travelled with their code (_logger, re, _SAME_FIELD_X_TOLERANCE_PX) - the same class of miss as _make_verifier, which ruff cannot flag. Counts unchanged: 81, 137, 14, 44, 57. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_011w2Pj4Ek3ZNMQVYcFHgXDY
128 lines
4.0 KiB
Python
128 lines
4.0 KiB
Python
"""Tests for saving generated images and the CLI entry point."""
|
|
|
|
from pathlib import Path
|
|
import tempfile
|
|
from unittest.mock import patch
|
|
|
|
from PIL import Image
|
|
|
|
from python_pkg.random_jpg.generate_jpeg import (
|
|
MAX_IMAGE_SIZE,
|
|
ImageConfig,
|
|
_save_image,
|
|
main,
|
|
)
|
|
|
|
|
|
class TestSaveImage:
|
|
"""Tests for _save_image function."""
|
|
|
|
def test_creates_output_folder(self) -> None:
|
|
"""Test that output folder is created if it doesn't exist."""
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
new_folder = Path(tmp_dir) / "new_subfolder"
|
|
image = Image.new("RGB", (10, 10))
|
|
config = ImageConfig(
|
|
size=10,
|
|
color_list=["#FF0000"],
|
|
block_size=10,
|
|
output_path="image.jpeg",
|
|
quality=90,
|
|
)
|
|
|
|
_save_image(image, config, 1, str(new_folder))
|
|
|
|
assert new_folder.exists()
|
|
|
|
def test_saves_with_correct_quality(self) -> None:
|
|
"""Test image is saved with specified quality."""
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
image = Image.new("RGB", (10, 10), color=(255, 0, 0))
|
|
config = ImageConfig(
|
|
size=10,
|
|
color_list=["#FF0000"],
|
|
block_size=10,
|
|
output_path="image.jpeg",
|
|
quality=50,
|
|
)
|
|
|
|
result_path = _save_image(image, config, 1, tmp_dir)
|
|
|
|
assert Path(result_path).exists()
|
|
|
|
|
|
class TestMain:
|
|
"""Tests for main CLI function."""
|
|
|
|
def test_main_generates_image_with_defaults(self) -> None:
|
|
"""Test main generates image with default arguments."""
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
patch("sys.argv", ["generate_jpeg.py"]),
|
|
patch(
|
|
"python_pkg.random_jpg.generate_jpeg.generate_bloated_jpeg"
|
|
) as mock_gen,
|
|
):
|
|
mock_gen.return_value = f"{tmp_dir}/test.jpeg"
|
|
main()
|
|
|
|
mock_gen.assert_called_once()
|
|
call_args = mock_gen.call_args
|
|
config = call_args[0][0]
|
|
assert config.size == 1000
|
|
assert config.block_size == 4
|
|
assert config.quality == 100
|
|
|
|
def test_main_respects_num_images_argument(self) -> None:
|
|
"""Test main generates multiple images when specified."""
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
patch("sys.argv", ["generate_jpeg.py", "-n", "3"]),
|
|
patch(
|
|
"python_pkg.random_jpg.generate_jpeg.generate_bloated_jpeg"
|
|
) as mock_gen,
|
|
):
|
|
mock_gen.return_value = f"{tmp_dir}/test.jpeg"
|
|
main()
|
|
|
|
assert mock_gen.call_count == 3
|
|
|
|
def test_main_uses_custom_size(self) -> None:
|
|
"""Test main respects custom size argument."""
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
patch("sys.argv", ["generate_jpeg.py", "-s", "500", "-b", "5"]),
|
|
patch(
|
|
"python_pkg.random_jpg.generate_jpeg.generate_bloated_jpeg"
|
|
) as mock_gen,
|
|
):
|
|
mock_gen.return_value = f"{tmp_dir}/test.jpeg"
|
|
main()
|
|
|
|
config = mock_gen.call_args[0][0]
|
|
assert config.size == 500
|
|
assert config.block_size == 5
|
|
|
|
def test_main_uses_custom_colors(self) -> None:
|
|
"""Test main respects custom color list."""
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
patch("sys.argv", ["generate_jpeg.py", "-c", "#AABBCC", "#112233"]),
|
|
patch(
|
|
"python_pkg.random_jpg.generate_jpeg.generate_bloated_jpeg"
|
|
) as mock_gen,
|
|
):
|
|
mock_gen.return_value = f"{tmp_dir}/test.jpeg"
|
|
main()
|
|
|
|
config = mock_gen.call_args[0][0]
|
|
assert config.color_list == ["#AABBCC", "#112233"]
|
|
|
|
|
|
class TestConstants:
|
|
"""Tests for module constants."""
|
|
|
|
def test_max_image_size(self) -> None:
|
|
"""Test MAX_IMAGE_SIZE constant value."""
|
|
assert MAX_IMAGE_SIZE == 1000
|