[OverlayLibrary] enable renaming all cells during add_source with rename_when='always'

masque/test/test_gdsii_lazy.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 import numpy
+import pytest
 from numpy.testing import assert_allclose
 
 from ..file import gdsii, gdsii_lazy
@@ -82,6 +83,35 @@ def test_gdsii_lazy_overlay_add_source_stays_lazy_for_processed_view(tmp_path: P
     assert set(abstract.ports) == {'A'}
 
 
+def test_gdsii_lazy_overlay_add_source_can_rename_every_source_cell() -> None:
+    src = _make_lazy_port_library()
+    overlay = gdsii_lazy.OverlayLibrary()
+
+    rename_map = overlay.add_source(
+        src,
+        rename_theirs=lambda _lib, name: f'mapped_{name}',
+        rename_when='always',
+    )
+
+    assert rename_map == {
+        'leaf': 'mapped_leaf',
+        'child': 'mapped_child',
+        'top': 'mapped_top',
+    }
+    assert tuple(overlay.keys()) == ('mapped_leaf', 'mapped_child', 'mapped_top')
+    assert 'mapped_leaf' in overlay['mapped_child'].refs
+
+
+def test_gdsii_lazy_overlay_add_source_rename_when_validation() -> None:
+    src = _make_lazy_port_library()
+
+    with pytest.raises(TypeError, match='rename_theirs'):
+        gdsii_lazy.OverlayLibrary().add_source(src, rename_when='always')
+
+    with pytest.raises(ValueError, match='rename mode'):
+        gdsii_lazy.OverlayLibrary().add_source(src, rename_when='sometimes')   # type: ignore[arg-type]
+
+
 def test_gdsii_lazy_processed_write_roundtrips_without_explicit_units(tmp_path: Path) -> None:
     gds_file = tmp_path / 'lazy_roundtrip.gds'
     src = _make_lazy_port_library()