From 0de0f0f95afbadd7408ff6094340f063d7264391 Mon Sep 17 00:00:00 2001 From: Stefan Wehrmeyer Date: Wed, 18 Dec 2013 13:02:18 +0100 Subject: [PATCH] Fix tests under Python 2.6 assertRaises is not a context manager under Py 2.6 --- test/test_persistence.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/test/test_persistence.py b/test/test_persistence.py index 20d18ba..1ae95fc 100644 --- a/test/test_persistence.py +++ b/test/test_persistence.py @@ -70,8 +70,7 @@ class DatabaseTestCase(unittest.TestCase): table.insert({'int_id': 124}) assert table.find_one(int_id = 123)['int_id'] == 123 assert table.find_one(int_id = 124)['int_id'] == 124 - with self.assertRaises(IntegrityError): - table.insert({'int_id': 123}) + self.assertRaises(IntegrityError, lambda: table.insert({'int_id': 123})) def test_create_table_shorthand1(self): pid = "int_id" @@ -84,8 +83,7 @@ class DatabaseTestCase(unittest.TestCase): table.insert({'int_id': 124}) assert table.find_one(int_id = 123)['int_id'] == 123 assert table.find_one(int_id = 124)['int_id'] == 124 - with self.assertRaises(IntegrityError): - table.insert({'int_id': 123}) + self.assertRaises(IntegrityError, lambda: table.insert({'int_id': 123})) def test_create_table_shorthand2(self): pid = "string_id"