remove context manager for assertRaises

alisaifee · Oct 5, 2015 · b204b45 · b204b45
1 parent a9830d8
commit b204b45
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_storage.py b/tests/test_storage.py
@@ -105,7 +105,7 @@ def test_redis_sentinel(self):
         self.assertTrue(limiter.hit(per_min))
 
     def test_pluggable_storage_invalid_construction(self):
-        with self.assertRaises(ConfigurationError):
+        def cons():
             class _(Storage):
                 def incr(self, key, expiry, elastic_expiry=False):
                     return
@@ -115,6 +115,7 @@ def get(self, key):
 
                 def get_expiry(self, key):
                     return time.time()
+        self.assertRaises(ConfigurationError, cons)
 
     def test_pluggable_storage_no_moving_window(self):
         class MyStorage(Storage):