85 lines
3 KiB
Python
85 lines
3 KiB
Python
from numpy.testing import assert_allclose
|
|
from numpy import pi
|
|
|
|
from ..abstract import Abstract
|
|
from ..ports import Port
|
|
from ..ref import Ref
|
|
|
|
|
|
def test_abstract_init() -> None:
|
|
ports = {"A": Port((0, 0), 0), "B": Port((10, 0), pi)}
|
|
abs_obj = Abstract("test", ports)
|
|
assert abs_obj.name == "test"
|
|
assert len(abs_obj.ports) == 2
|
|
assert abs_obj.ports["A"] is not ports["A"] # Should be deepcopied
|
|
|
|
|
|
def test_abstract_transform() -> None:
|
|
abs_obj = Abstract("test", {"A": Port((10, 0), 0)})
|
|
# Rotate 90 deg around (0,0)
|
|
abs_obj.rotate_around((0, 0), pi / 2)
|
|
# (10, 0) rot 0 -> (0, 10) rot pi/2
|
|
assert_allclose(abs_obj.ports["A"].offset, [0, 10], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, pi / 2, atol=1e-10)
|
|
|
|
# Mirror across x axis (axis 0): flips y-offset
|
|
abs_obj.mirror(0)
|
|
# (0, 10) mirrored(0) -> (0, -10)
|
|
# rotation pi/2 mirrored(0) -> -pi/2 == 3pi/2
|
|
assert_allclose(abs_obj.ports["A"].offset, [0, -10], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, 3 * pi / 2, atol=1e-10)
|
|
|
|
|
|
def test_abstract_ref_transform() -> None:
|
|
abs_obj = Abstract("test", {"A": Port((10, 0), 0)})
|
|
ref = Ref(offset=(100, 100), rotation=pi / 2, mirrored=True)
|
|
|
|
# Apply ref transform
|
|
abs_obj.apply_ref_transform(ref)
|
|
# Ref order: mirror, rotate, scale, translate
|
|
|
|
# 1. mirror (across x: y -> -y)
|
|
# (10, 0) rot 0 -> (10, 0) rot 0
|
|
|
|
# 2. rotate pi/2 around (0,0)
|
|
# (10, 0) rot 0 -> (0, 10) rot pi/2
|
|
|
|
# 3. translate (100, 100)
|
|
# (0, 10) -> (100, 110)
|
|
|
|
assert_allclose(abs_obj.ports["A"].offset, [100, 110], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, pi / 2, atol=1e-10)
|
|
|
|
|
|
def test_abstract_ref_transform_scales_offsets() -> None:
|
|
abs_obj = Abstract("test", {"A": Port((10, 0), 0)})
|
|
ref = Ref(offset=(100, 100), rotation=pi / 2, mirrored=True, scale=2)
|
|
|
|
abs_obj.apply_ref_transform(ref)
|
|
|
|
assert_allclose(abs_obj.ports["A"].offset, [100, 120], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, pi / 2, atol=1e-10)
|
|
|
|
|
|
def test_abstract_undo_transform() -> None:
|
|
abs_obj = Abstract("test", {"A": Port((100, 110), pi / 2)})
|
|
ref = Ref(offset=(100, 100), rotation=pi / 2, mirrored=True)
|
|
|
|
abs_obj.undo_ref_transform(ref)
|
|
assert_allclose(abs_obj.ports["A"].offset, [10, 0], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, 0, atol=1e-10)
|
|
|
|
|
|
def test_abstract_undo_transform_scales_offsets() -> None:
|
|
abs_obj = Abstract("test", {"A": Port((100, 120), pi / 2)})
|
|
ref = Ref(offset=(100, 100), rotation=pi / 2, mirrored=True, scale=2)
|
|
|
|
abs_obj.undo_ref_transform(ref)
|
|
assert_allclose(abs_obj.ports["A"].offset, [10, 0], atol=1e-10)
|
|
assert abs_obj.ports["A"].rotation is not None
|
|
assert_allclose(abs_obj.ports["A"].rotation, 0, atol=1e-10)
|