Organize tools and move Python tests into tools/tests
This commit is contained in:
zwitschi committed
1 parent
0aacee6518
commit
e4e1566e55
22 files changed
+209
-3
No files matched your search
@@ -0,0 +1,207 @@
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch
|
||||
|
||||
from PIL import Image, ImageDraw
|
||||
|
||||
import asset_pipeline
|
||||
|
||||
|
||||
class AssetPipelineTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.temp_dir = tempfile.TemporaryDirectory()
|
||||
self.root = Path(self.temp_dir.name)
|
||||
|
||||
def tearDown(self):
|
||||
self.temp_dir.cleanup()
|
||||
|
||||
def test_crop_uses_center_and_exact_dimensions(self):
|
||||
source = self.root / "source.png"
|
||||
output = self.root / "crop.png"
|
||||
Image.new("RGB", (80, 60), (20, 40, 60)).save(source)
|
||||
|
||||
asset_pipeline.crop_image(source, output, 32, 24)
|
||||
|
||||
with Image.open(output) as image:
|
||||
self.assertEqual(image.size, (32, 24))
|
||||
|
||||
def test_resize_contain_preserves_aspect_and_exact_canvas(self):
|
||||
source = self.root / "source.png"
|
||||
output = self.root / "resized.png"
|
||||
Image.new("RGB", (80, 40), (20, 40, 60)).save(source)
|
||||
|
||||
asset_pipeline.resize_image(source, output, 64, 64, "contain")
|
||||
|
||||
with Image.open(output) as image:
|
||||
self.assertEqual(image.size, (64, 64))
|
||||
self.assertEqual(image.getpixel((32, 0)), (0, 0, 0, 0))
|
||||
|
||||
def test_texture_audit_accepts_valid_tile_and_rejects_bad_seams(self):
|
||||
good = self.root / "good.png"
|
||||
bad = self.root / "bad.png"
|
||||
Image.new("RGB", (512, 512), (80, 80, 80)).save(good)
|
||||
mismatch = Image.new("RGB", (512, 512), (80, 80, 80))
|
||||
for y in range(512):
|
||||
mismatch.putpixel((0, y), (255, 0, 0))
|
||||
mismatch.save(bad)
|
||||
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
good, "tile")["quality"], "good")
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
bad, "tile")["quality"], "bad")
|
||||
|
||||
def test_seamless_transform_makes_tile_edges_continuous(self):
|
||||
image_path = self.root / "tile.png"
|
||||
output = self.root / "seamless.png"
|
||||
image = Image.new("RGB", (512, 512))
|
||||
for x in range(512):
|
||||
for y in range(512):
|
||||
image.putpixel((x, y), (x // 2, y // 2, (x + y) // 4))
|
||||
image.save(image_path)
|
||||
|
||||
asset_pipeline.seamless_image(image_path, output)
|
||||
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
output, "tile")["quality"], "good")
|
||||
|
||||
def test_tile_generation_applies_seamless_transform(self):
|
||||
output = self.root / "generated_tile.png"
|
||||
args = type("Args", (), {
|
||||
"prompt": "brick", "role": "tile", "output": str(output),
|
||||
"width": None, "height": None, "transparent": False,
|
||||
"model": "test-model", "request_size": 1024,
|
||||
})()
|
||||
generated = Image.new("RGB", (512, 512), (80, 80, 80))
|
||||
for y in range(512):
|
||||
generated.putpixel((0, y), (255, 0, 0))
|
||||
|
||||
with patch.object(asset_pipeline.generate_textures, "generate_image",
|
||||
return_value=generated):
|
||||
asset_pipeline._generate(args)
|
||||
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
output, "tile")["quality"], "good")
|
||||
|
||||
def test_background_audit_rejects_small_image(self):
|
||||
image_path = self.root / "small.png"
|
||||
Image.new("RGB", (640, 360)).save(image_path)
|
||||
|
||||
result = asset_pipeline.audit_image(image_path, "background")
|
||||
|
||||
self.assertEqual(result["quality"], "bad")
|
||||
self.assertTrue(
|
||||
any("too small" in error for error in result["errors"]))
|
||||
|
||||
def test_grayscale_vignette_is_valid_without_alpha(self):
|
||||
image_path = self.root / "vignette.png"
|
||||
image = Image.new("L", (512, 512), 0)
|
||||
draw = ImageDraw.Draw(image)
|
||||
for radius in range(255, -1, -1):
|
||||
value = int(255 * (1 - radius / 255))
|
||||
draw.ellipse((256 - radius, 256 - radius,
|
||||
256 + radius, 256 + radius), fill=value)
|
||||
image.save(image_path)
|
||||
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
image_path, "overlay")["quality"], "good")
|
||||
|
||||
def test_overlay_audit_rejects_blank_transparent_image(self):
|
||||
image_path = self.root / "blank_overlay.png"
|
||||
Image.new("RGBA", (512, 512), (0, 0, 0, 0)).save(image_path)
|
||||
|
||||
result = asset_pipeline.audit_image(image_path, "overlay")
|
||||
|
||||
self.assertEqual(result["quality"], "bad")
|
||||
self.assertTrue(any("near-blank frame" in error
|
||||
for error in result["errors"]))
|
||||
|
||||
def test_icon_audit_rejects_opaque_rgba_images(self):
|
||||
image_path = self.root / "opaque_icon.png"
|
||||
Image.new("RGBA", (64, 64), (255, 255, 255, 255)).save(image_path)
|
||||
|
||||
result = asset_pipeline.audit_image(image_path, "icon")
|
||||
|
||||
self.assertEqual(result["quality"], "bad")
|
||||
self.assertTrue(any("transparency required" in error
|
||||
for error in result["errors"]))
|
||||
|
||||
def test_backdrop_audit_requires_transparency(self):
|
||||
image_path = self.root / "skyline.png"
|
||||
Image.new("RGB", (1024, 256), (20, 30, 40)).save(image_path)
|
||||
|
||||
result = asset_pipeline.audit_image(image_path, "backdrop")
|
||||
|
||||
self.assertEqual(result["quality"], "bad")
|
||||
self.assertTrue(any("transparency required" in error
|
||||
for error in result["errors"]))
|
||||
|
||||
def test_generation_reuses_existing_backend_and_targets_role_dimensions(self):
|
||||
output = self.root / "background.png"
|
||||
args = type("Args", (), {
|
||||
"prompt": "Hollywood at night", "role": "background", "output": str(output),
|
||||
"width": 1920, "height": 1080, "transparent": None,
|
||||
"model": "test-model", "request_size": 2048,
|
||||
})()
|
||||
generated = Image.new("RGB", (256, 256), (20, 40, 60))
|
||||
|
||||
with patch.object(asset_pipeline.generate_textures, "generate_image",
|
||||
return_value=generated) as backend:
|
||||
asset_pipeline._generate(args)
|
||||
|
||||
self.assertIn("Hollywood at night", backend.call_args.args[0])
|
||||
self.assertIn("wide 16:9", backend.call_args.args[0])
|
||||
self.assertEqual(backend.call_args.kwargs["aspect_ratio"], "3:2")
|
||||
with Image.open(output) as image:
|
||||
self.assertEqual(image.size, (1920, 1080))
|
||||
|
||||
def test_generation_can_recolor_transparent_glyph(self):
|
||||
output = self.root / "icon.png"
|
||||
args = type("Args", (), {
|
||||
"prompt": "flashlight", "role": "icon", "output": str(output),
|
||||
"width": None, "height": None, "transparent": True,
|
||||
"model": "test-model", "request_size": 1024,
|
||||
"foreground": (255, 255, 255, 255),
|
||||
})()
|
||||
generated = Image.new("RGBA", (64, 64), (0, 0, 0, 0))
|
||||
generated.putpixel((32, 32), (20, 20, 20, 255))
|
||||
|
||||
with patch.object(asset_pipeline.generate_textures, "generate_image",
|
||||
return_value=generated):
|
||||
asset_pipeline._generate(args)
|
||||
|
||||
with Image.open(output) as image:
|
||||
self.assertEqual(image.getpixel((32, 32)), (255, 255, 255, 255))
|
||||
self.assertEqual(image.getpixel((0, 0))[3], 0)
|
||||
|
||||
def test_screenshot_audit_accepts_game_frame_and_rejects_blank_capture(self):
|
||||
valid = self.root / "frame.png"
|
||||
blank = self.root / "blank.png"
|
||||
frame = Image.new("RGB", (1280, 720), (0, 0, 0))
|
||||
ImageDraw.Draw(frame).rectangle(
|
||||
(640, 0, 1279, 719), fill=(255, 255, 255))
|
||||
frame.save(valid)
|
||||
Image.new("RGB", (1280, 720), (0, 0, 0)).save(blank)
|
||||
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
valid, "screenshot")["quality"], "good")
|
||||
self.assertEqual(asset_pipeline.audit_image(
|
||||
blank, "screenshot")["quality"], "bad")
|
||||
|
||||
def test_role_inference_uses_game_asset_paths(self):
|
||||
self.assertEqual(asset_pipeline.infer_role(
|
||||
Path("assets/textures/icons/wand.png")), "icon")
|
||||
self.assertEqual(asset_pipeline.infer_role(
|
||||
Path("assets/textures/cards/ticket.png")), "card")
|
||||
self.assertEqual(asset_pipeline.infer_role(
|
||||
Path("assets/textures/skyline.png")), "backdrop")
|
||||
|
||||
def test_role_generator_defaults_to_validated_image_model(self):
|
||||
args = asset_pipeline.build_parser().parse_args([
|
||||
"generate", "--prompt", "test", "--role", "tile", "--output", "tile.png"])
|
||||
|
||||
self.assertEqual(args.model, asset_pipeline.DEFAULT_ROLE_IMAGE_MODEL)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user