diff options
| author | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2011-01-08 00:40:27 +0000 |
|---|---|---|
| committer | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2011-01-08 23:54:48 +0000 |
| commit | f8c4335f3590d8d254fcabcb7b133bd93091e8cf (patch) | |
| tree | b411a8aa04c5932381226729a48a95b2b393ff76 /tests/test_lobject.py | |
| parent | 43c8fce45c858e96cb1cbb24d2623d721bbbf51c (diff) | |
| download | psycopg2-f8c4335f3590d8d254fcabcb7b133bd93091e8cf.tar.gz | |
Avoid ResourceWarning in tests in Python 3.2
Diffstat (limited to 'tests/test_lobject.py')
| -rw-r--r-- | tests/test_lobject.py | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/tests/test_lobject.py b/tests/test_lobject.py index 1c71fb4..cc6621e 100644 --- a/tests/test_lobject.py +++ b/tests/test_lobject.py @@ -182,7 +182,11 @@ class LargeObjectTests(LargeObjectMixin, unittest.TestCase): filename = os.path.join(self.tmpdir, "data.txt") lo.export(filename) self.assertTrue(os.path.exists(filename)) - self.assertEqual(open(filename, "rb").read(), b("some data")) + f = open(filename, "rb") + try: + self.assertEqual(f.read(), b("some data")) + finally: + f.close() def test_close_twice(self): lo = self.conn.lobject() @@ -224,7 +228,11 @@ class LargeObjectTests(LargeObjectMixin, unittest.TestCase): filename = os.path.join(self.tmpdir, "data.txt") lo.export(filename) self.assertTrue(os.path.exists(filename)) - self.assertEqual(open(filename, "rb").read(), b("some data")) + f = open(filename, "rb") + try: + self.assertEqual(f.read(), b("some data")) + finally: + f.close() def test_close_after_commit(self): lo = self.conn.lobject() @@ -279,7 +287,11 @@ class LargeObjectTests(LargeObjectMixin, unittest.TestCase): filename = os.path.join(self.tmpdir, "data.txt") lo.export(filename) self.assertTrue(os.path.exists(filename)) - self.assertEqual(open(filename, "rb").read(), b("some data")) + f = open(filename, "rb") + try: + self.assertEqual(f.read(), b("some data")) + finally: + f.close() decorate_all_tests(LargeObjectTests, skip_if_no_lo) decorate_all_tests(LargeObjectTests, skip_if_green) |