commit aad15414842ae797fa3f0d7a4a2868e9f8f71370
author Ezio Melotti <ezio.melotti@gmail.com> Tue Aug 03 06:49:14 2010 +0000
committer Ezio Melotti <ezio.melotti@gmail.com> Tue Aug 03 06:49:14 2010 +0000
tree 84876e8b9b6706efc4dbf6bc4ef07e731da69757
parent 192f6782c6aa1cb1979aa6c494e01268f5aabab1

Fix deprecation warnings in test_os.py

Lib/test/test_os.py

diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
index 3cb411b..e54ab4b 100644
--- a/Lib/test/test_os.py
+++ b/Lib/test/test_os.py
@@ -500,14 +500,16 @@
 class URandomTests (unittest.TestCase):
     def test_urandom(self):
         try:
-            self.assertEqual(len(os.urandom(1)), 1)
-            self.assertEqual(len(os.urandom(10)), 10)
-            self.assertEqual(len(os.urandom(100)), 100)
-            self.assertEqual(len(os.urandom(1000)), 1000)
-            # see http://bugs.python.org/issue3708
-            self.assertEqual(len(os.urandom(0.9)), 0)
-            self.assertEqual(len(os.urandom(1.1)), 1)
-            self.assertEqual(len(os.urandom(2.0)), 2)
+            with test_support._check_py3k_warnings(
+                ('integer argument expected, got float', DeprecationWarning)):
+                self.assertEqual(len(os.urandom(1)), 1)
+                self.assertEqual(len(os.urandom(10)), 10)
+                self.assertEqual(len(os.urandom(100)), 100)
+                self.assertEqual(len(os.urandom(1000)), 1000)
+                # see http://bugs.python.org/issue3708
+                self.assertEqual(len(os.urandom(0.9)), 0)
+                self.assertEqual(len(os.urandom(1.1)), 1)
+                self.assertEqual(len(os.urandom(2.0)), 2)
         except NotImplementedError:
             pass