cpu_affinity() test: use addCleanup to reset initial value

1 files changed, 1 insertions, 0 deletions

diff --git a/psutil/tests/test_process.py b/psutil/tests/test_process.py
index 1a95bc19..50e0cc74 100755
--- a/psutil/tests/test_process.py
+++ b/psutil/tests/test_process.py
@@ -845,6 +845,7 @@ class TestProcess(unittest.TestCase):
     def test_cpu_affinity(self):
         p = psutil.Process()
         initial = p.cpu_affinity()
+        self.addCleanup(p.cpu_affinity, initial)
         if hasattr(os, "sched_getaffinity"):
             self.assertEqual(initial, list(os.sched_getaffinity(p.pid)))
         self.assertEqual(len(initial), len(set(initial)))