Change self.assertRaises to context managers.

1 files changed, 8 insertions, 4 deletions

diff --git a/tests/test_bio.py b/tests/test_bio.py
index 88b6902..ab727d5 100644
--- a/tests/test_bio.py
+++ b/tests/test_bio.py
@@ -41,9 +41,12 @@ class CipherStreamTestCase(unittest.TestCase):
         cf.close()
         assert not cf.readable()
         
-        self.assertRaises(IOError, cf.read)
-        self.assertRaises(IOError, cf.readline)
-        self.assertRaises(IOError, cf.readlines)
+        with self.assertRaises(IOError):
+            cf.read()
+        with self.assertRaises(IOError):
+            cf.readline()
+        with self.assertRaises(IOError):
+            cf.readlines()
     
         assert data == data2, '%s algorithm cipher test failed' % algo
         
@@ -65,7 +68,8 @@ class CipherStreamTestCase(unittest.TestCase):
         for i in ciphers:
             self.try_algo(i)
 
-        self.assertRaises(ValueError, self.try_algo, 'nosuchalgo4567')
+        with self.assertRaises(ValueError):
+            self.try_algo('nosuchalgo4567')
 
 def suite():
     suite = unittest.TestSuite()