try to fix failure on py 3

mrjefftang · Aug 7, 2015 · 3e5aa8a · 3e5aa8a
1 parent a347904
commit 3e5aa8a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_psutil.py b/test/test_psutil.py
@@ -1514,7 +1514,8 @@ def test_rlimit(self):
             with self.assertRaises(IOError) as exc:
                 with open(TESTFN, "wb") as f:
                     f.write(b"X" * 1025)
-            self.assertEqual(exc.exception[0], errno.EFBIG)
+            self.assertEqual(exc.exception.errno if PY3 else exc.exception[0],
+                             errno.EFBIG)
         finally:
             p.rlimit(psutil.RLIMIT_FSIZE, (soft, hard))
             self.assertEqual(p.rlimit(psutil.RLIMIT_FSIZE), (soft, hard))