test: fix test_xlib_display messing with pointer mapping

benoit-pierre · benoit-pierre · commit 4bd0b6d7df0c · 2016-06-27T12:37:20.000+02:00
diff --git a/test/test_xlib_display.py b/test/test_xlib_display.py
@@ -40,9 +40,13 @@ def test_pointer_mapping_is_list(self):
         self.assertTrue(isinstance(self.display.get_pointer_mapping(), list))
 
     def test_set_get_pointer_mapping(self):
-        length = len(self.display.get_pointer_mapping())
-        self.display.set_pointer_mapping([0] * length)
-        self.assertEqual(self.display.get_pointer_mapping(), [0] * length)
+        orig_mapping = self.display.get_pointer_mapping()
+        length = len(orig_mapping)
+        try:
+            self.display.set_pointer_mapping([0] * length)
+            self.assertEqual(self.display.get_pointer_mapping(), [0] * length)
+        finally:
+            self.display.set_pointer_mapping(orig_mapping)
 
     def test_can_close_display(self):
         self.display.close()
