[tests] Add machine-generated test suite

masque/test/test_abstract.py

@@ -0,0 +1,58 @@
+import pytest
+import numpy
+from numpy.testing import assert_equal, assert_allclose
+from numpy import pi
+
+from ..abstract import Abstract
+from ..ports import Port
+from ..ref import Ref
+
+def test_abstract_init():
+    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():
+    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_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_allclose(abs_obj.ports["A"].rotation, 3*pi/2, atol=1e-10)
+
+def test_abstract_ref_transform():
+    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_allclose(abs_obj.ports["A"].rotation, pi/2, atol=1e-10)
+
+def test_abstract_undo_transform():
+    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_allclose(abs_obj.ports["A"].rotation, 0, atol=1e-10)