51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from ..utils.pack2d import maxrects_bssf, pack_patterns
|
|
from ..library import Library
|
|
from ..pattern import Pattern
|
|
|
|
|
|
def test_maxrects_bssf_simple() -> None:
|
|
# Pack two 10x10 squares into one 20x10 container
|
|
rects = [[10, 10], [10, 10]]
|
|
containers = [[0, 0, 20, 10]]
|
|
|
|
locs, rejects = maxrects_bssf(rects, containers)
|
|
|
|
assert not rejects
|
|
# They should be at (0,0) and (10,0)
|
|
assert {tuple(loc) for loc in locs} == {(0.0, 0.0), (10.0, 0.0)}
|
|
|
|
|
|
def test_maxrects_bssf_reject() -> None:
|
|
# Try to pack a too-large rectangle
|
|
rects = [[10, 10], [30, 30]]
|
|
containers = [[0, 0, 20, 20]]
|
|
|
|
locs, rejects = maxrects_bssf(rects, containers, allow_rejects=True)
|
|
assert 1 in rejects # Second rect rejected
|
|
assert 0 not in rejects
|
|
|
|
|
|
def test_pack_patterns() -> None:
|
|
lib = Library()
|
|
p1 = Pattern()
|
|
p1.polygon((1, 0), vertices=[[0, 0], [10, 0], [10, 10], [0, 10]])
|
|
lib["p1"] = p1
|
|
|
|
p2 = Pattern()
|
|
p2.polygon((1, 0), vertices=[[0, 0], [5, 0], [5, 5], [0, 5]])
|
|
lib["p2"] = p2
|
|
|
|
# Containers: one 20x20
|
|
containers = [[0, 0, 20, 20]]
|
|
# 2um spacing
|
|
pat, rejects = pack_patterns(lib, ["p1", "p2"], containers, spacing=(2, 2))
|
|
|
|
assert not rejects
|
|
assert len(pat.refs) == 2
|
|
assert "p1" in pat.refs
|
|
assert "p2" in pat.refs
|
|
|
|
# Check that they don't overlap (simple check via bounds)
|
|
# p1 size 10x10, effectively 12x12
|
|
# p2 size 5x5, effectively 7x7
|
|
# Both should fit in 20x20
|