using python 2.6 compatible way of assertRaises

malderete · Nov 30, 2014 · 1cac35b · 1cac35b
1 parent 6d48100
commit 1cac35b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/fields/fields.py b/tests/fields/fields.py
@@ -3076,8 +3076,10 @@ def test_undefined_field_exception(self):
         class Doc(Document):
             foo = StringField(db_field='f')
 
-        with self.assertRaises(FieldDoesNotExist):
+        def test():
             Doc(bar='test')
 
+        self.assertRaises(FieldDoesNotExist, test)
+
 if __name__ == '__main__':
     unittest.main()