Fix examples

[linpy.git] / pypol / tests / test_domains.py

diff --git a/pypol/tests/test_domains.py b/pypol/tests/test_domains.py
index 55853fd..e9dd1cd 100644 (file)
--- a/pypol/tests/test_domains.py
+++ b/pypol/tests/test_domains.py
@@ -1,7 +1,7 @@
 import unittest
 
 from ..domains import *
-from ..linexprs import symbols
+from ..linexprs import Symbol, symbols
 from ..polyhedra import *
 
 
@@ -78,10 +78,10 @@ class TestDomain(unittest.TestCase):
         self.assertEqual(self.square1.polyhedral_hull(), self.hull)
 
     def test_project_out(self):
-        self.assertEqual(self.square1.project_out('x'), self.dropped)
-        self.assertEqual(self.square1.project_out('x y'), self.universe)
-        self.assertEqual(self.universe.project_out(' '), self.universe)
-        self.assertEqual(self.empty.project_out(' '), Empty)
+        self.assertEqual(self.square1.project_out(symbols('x')), self.dropped)
+        self.assertEqual(self.square1.project_out(symbols('x y')), self.universe)
+        self.assertEqual(self.universe.project_out([]), self.universe)
+        self.assertEqual(self.empty.project_out([]), Empty)
 
     def test_simplify(self):
         self.assertEqual(self.universe.simplify(), self.universe)