Add basic tests for drop_dims

[linpy.git] / pypol / domains.py

diff --git a/pypol/domains.py b/pypol/domains.py
index 4a1f9aa..be87ffb 100644 (file)
--- a/pypol/domains.py
+++ b/pypol/domains.py
@@ -154,15 +154,18 @@ class Domain:
 
     def drop_dims(self, dims):
         # use to remove certain variables use isl_set_drop_constraints_involving_dims instead?
 
     def drop_dims(self, dims):
         # use to remove certain variables use isl_set_drop_constraints_involving_dims instead?

-        from .polyhedra import Polyhedron       
+        from .polyhedra import Polyhedron      

         dims = list(dims)
         symbols = list(self.symbols)
         islset = self._toislset(self.polyhedra, self.symbols)
         for dim in dims:
             if dim in symbols:
         dims = list(dims)
         symbols = list(self.symbols)
         islset = self._toislset(self.polyhedra, self.symbols)
         for dim in dims:
             if dim in symbols:

-                num = symbols.index(dim)
-                islbset = libisl.isl_set_drop_constraints_involving_dims(islset, libisl.isl_dim_set, num, 1)
-        return Polyhedron._fromislset(islbset, self.symbols)
+                first = symbols.index(dim)
+                islbset = libisl.isl_set_project_out(islset, libisl.isl_dim_set, first, 1)  
+                symbols.__delitem__(first)     
+            else:
+                islbset = libisl.isl_set_project_out(islset, libisl.isl_dim_set, 0, 0)        
+        return Polyhedron._fromislset(islbset, symbols)

     
     def sample(self):
         from .polyhedra import Polyhedron
     
     def sample(self):
         from .polyhedra import Polyhedron