mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 18:03:07 +02:00
Krzysztof kuhy Rudnicki
5ef944abc9
- download_cats: Test generate_cats functionality (100% coverage) - keyboard_coop: Test keyboard_listener with mocked pynput (22% coverage) - mock_server: Test mitmproxy request interceptor (100% coverage) - random_jpg: Test JPEG generation with mocked Pillow (100% coverage) - randomize_numbers: Test random_digits functions (99% coverage) - scrape_website: Test scrape_comics with mocked requests (98% coverage) - split: Test text splitting utilities (100% coverage) - tag_divider: Test tag_divider with mock filesystem (44% coverage) - extract_links: Add HTML fixture for tests
252 lines
8.1 KiB
Python
252 lines
8.1 KiB
Python
"""Unit tests for generate_jpeg module."""
|
|
|
|
from pathlib import Path
|
|
import tempfile
|
|
from unittest.mock import patch
|
|
|
|
from PIL import Image
|
|
import pytest
|
|
|
|
from python_pkg.random_jpg.generate_jpeg import (
|
|
MAX_IMAGE_SIZE,
|
|
ImageConfig,
|
|
_create_random_image,
|
|
_save_image,
|
|
generate_bloated_jpeg,
|
|
main,
|
|
)
|
|
|
|
|
|
class TestImageConfig:
|
|
"""Tests for ImageConfig dataclass."""
|
|
|
|
def test_creates_config_with_all_fields(self) -> None:
|
|
"""Test ImageConfig stores all configuration fields."""
|
|
config = ImageConfig(
|
|
size=100,
|
|
color_list=["#FF0000", "#00FF00"],
|
|
block_size=10,
|
|
output_path="test.jpeg",
|
|
quality=95,
|
|
)
|
|
assert config.size == 100
|
|
assert config.color_list == ["#FF0000", "#00FF00"]
|
|
assert config.block_size == 10
|
|
assert config.output_path == "test.jpeg"
|
|
assert config.quality == 95
|
|
|
|
|
|
class TestGenerateBloatedJpeg:
|
|
"""Tests for generate_bloated_jpeg function."""
|
|
|
|
def test_generates_image_file(self) -> None:
|
|
"""Test that generate_bloated_jpeg creates an image file."""
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
config = ImageConfig(
|
|
size=100,
|
|
color_list=["#FF0000", "#00FF00", "#0000FF"],
|
|
block_size=10,
|
|
output_path="test.jpeg",
|
|
quality=90,
|
|
)
|
|
result_path = generate_bloated_jpeg(config, 1, tmp_dir)
|
|
|
|
assert Path(result_path).exists()
|
|
# Verify it's a valid image
|
|
with Image.open(result_path) as img:
|
|
assert img.size == (100, 100)
|
|
|
|
def test_raises_error_for_size_exceeding_max(self) -> None:
|
|
"""Test ValueError when size exceeds MAX_IMAGE_SIZE."""
|
|
config = ImageConfig(
|
|
size=MAX_IMAGE_SIZE + 1,
|
|
color_list=["#FF0000"],
|
|
block_size=10,
|
|
output_path="test.jpeg",
|
|
quality=90,
|
|
)
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
pytest.raises(ValueError, match="1000 pixels or less"),
|
|
):
|
|
generate_bloated_jpeg(config, 1, tmp_dir)
|
|
|
|
def test_raises_error_for_indivisible_size(self) -> None:
|
|
"""Test ValueError when size not divisible by block_size."""
|
|
config = ImageConfig(
|
|
size=100,
|
|
color_list=["#FF0000"],
|
|
block_size=7, # 100 is not divisible by 7
|
|
output_path="test.jpeg",
|
|
quality=90,
|
|
)
|
|
with (
|
|
tempfile.TemporaryDirectory() as tmp_dir,
|
|
pytest.raises(ValueError, match="divisible by block_size"),
|
|
):
|
|
generate_bloated_jpeg(config, 1, tmp_dir)
|
|
|
|
def test_unique_naming_with_index(self) -> None:
|
|
"""Test that images are named with index."""
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
config = ImageConfig(
|
|
size=100,
|
|
color_list=["#FF0000"],
|
|
block_size=10,
|
|
output_path="output.jpeg",
|
|
quality=90,
|
|
)
|
|
path1 = generate_bloated_jpeg(config, 1, tmp_dir)
|
|
path2 = generate_bloated_jpeg(config, 2, tmp_dir)
|
|
|
|
assert "output_1.jpeg" in path1
|
|
assert "output_2.jpeg" in path2
|
|
|
|
|
|
class TestCreateRandomImage:
|
|
"""Tests for _create_random_image function."""
|
|
|
|
def test_creates_image_with_correct_size(self) -> None:
|
|
"""Test created image has correct dimensions."""
|
|
config = ImageConfig(
|
|
size=100,
|
|
color_list=["#FF0000", "#00FF00"],
|
|
block_size=10,
|
|
output_path="test.jpeg",
|
|
quality=90,
|
|
)
|
|
image = _create_random_image(config)
|
|
|
|
assert image.size == (100, 100)
|
|
assert image.mode == "RGB"
|
|
|
|
def test_fills_image_with_blocks(self) -> None:
|
|
"""Test image is filled with colored blocks."""
|
|
config = ImageConfig(
|
|
size=20,
|
|
color_list=["#FF0000"], # Only red
|
|
block_size=10,
|
|
output_path="test.jpeg",
|
|
quality=90,
|
|
)
|
|
image = _create_random_image(config)
|
|
pixels = image.load()
|
|
|
|
# With only red color, all pixels should be red
|
|
for x in range(20):
|
|
for y in range(20):
|
|
assert pixels[x, y] == (255, 0, 0)
|
|
|
|
|
|
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
|