nan
196b964efc
- Implemented `cutout_shell.py` to build front normals and shell positions for balloon geometry. - Created `cutout_volume.py` to establish smooth center fields and manage balloon surface attributes. - Introduced `pisa_sr_export.py` for exporting PiSA-SR components with dependencies and asset finalization.
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
import bpy
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
sys.path.insert(0, str(Path(__file__).parents[1] / "src"))
|
|
from anyimage.operators.upscale import UpscaleImage
|
|
from anyimage.common.image import image_pixels
|
|
|
|
|
|
def test_upscale_result_replacement_undo_restores_source(tmp_path):
|
|
source = bpy.data.images.new("Upscale undo source", width=8, height=8, alpha=True)
|
|
source.pixels.foreach_set(np.ones(8*8*4, dtype=np.float32))
|
|
source.pack()
|
|
owner = bpy.data.objects.new("Upscale undo owner", None)
|
|
owner.empty_display_type = "IMAGE"
|
|
owner.data = source
|
|
bpy.context.scene.collection.objects.link(owner)
|
|
owner_name, source_name = owner.name, source.name
|
|
before = image_pixels(source)
|
|
Image.new("RGBA", (16,16), (32,64,96,128)).save(tmp_path / "upscale.0001.png")
|
|
|
|
def execute(operator, context):
|
|
operator.source_object_name = owner_name
|
|
operator.source_image_name = source_name
|
|
operator.model = "PISA_SR"
|
|
operator.response(context, SimpleNamespace(directory=tmp_path))
|
|
return {"FINISHED"}
|
|
|
|
bpy.context.preferences.edit.use_global_undo = True
|
|
with patch.object(UpscaleImage, "execute", execute), patch.object(
|
|
UpscaleImage, "poll", classmethod(lambda cls, context: True)
|
|
):
|
|
bpy.utils.register_class(UpscaleImage)
|
|
try:
|
|
bpy.ops.ed.undo_push(message="Before Upscale")
|
|
assert bpy.ops.anyimage.upscale_image("EXEC_DEFAULT", True) == {"FINISHED"}
|
|
assert tuple(bpy.data.objects[owner_name].data.size) == (16,16)
|
|
assert bpy.ops.ed.undo() == {"FINISHED"}
|
|
restored = bpy.data.objects[owner_name].data
|
|
assert tuple(restored.size) == (8,8)
|
|
np.testing.assert_array_equal(image_pixels(restored), before)
|
|
finally:
|
|
bpy.utils.unregister_class(UpscaleImage)
|
|
bpy.data.objects.remove(bpy.data.objects[owner_name], do_unlink=True)
|
|
bpy.data.images.remove(bpy.data.images[source_name])
|