add namedisl.Map.coalesce()

Author: a-alveyblanc

namedisl/set_like.py

@@ -553,6 +553,7 @@ class Map(_NamedIslMapLike[isl.Map], _NamedIslUnbasic[isl.Map]):
     .. automethod:: complement
     .. automethod:: convex_hull
     .. automethod:: get_basic_maps
+    .. automethod:: coalesce
     .. automethod:: domain
     .. automethod:: range
     .. automethod:: intersect_domain
@@ -574,6 +575,9 @@ def convex_hull(self) -> BasicMap:
     def get_basic_maps(self) -> list[BasicMap]:
         return [BasicMap(bs, self.space) for bs in self._obj.get_basic_maps()]
 
+    def coalesce(self) -> Self:
+        return type(self)(self._obj.coalesce(), self.space)
+
     def domain(self) -> Set:
         return Set(
             self._obj.domain(),

namedisl/test/test_set_like.py

@@ -307,6 +307,14 @@ def test_map_from_map() -> None:
     print(named_map)
 
 
+def test_map_coalesce() -> None:
+    map_ = nisl.make_map(
+        "{ [i] -> [j = i] : 0 <= i < 5 or 5 <= i < 10 }"
+    )
+    assert len(map_.get_basic_maps()) == 2
+    assert len(map_.coalesce().get_basic_maps()) == 1
+
+
 @pytest.mark.parametrize("ndims_domain", [2, 3, 4, 5])
 @pytest.mark.parametrize("ndims_range", [2, 3, 4, 5])
 @pytest.mark.parametrize("has_params", [True, False])