From 0f4fd0d828d5e950bb9913e42597774aea0512df Mon Sep 17 00:00:00 2001 From: Daniele Varrazzo Date: Wed, 19 Oct 2011 02:25:53 +0100 Subject: [PATCH] Test suite fixed to make it compatible with the ctypes implementation --- tests/test_dates.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/tests/test_dates.py b/tests/test_dates.py index 27abbc1e..aa6e501f 100755 --- a/tests/test_dates.py +++ b/tests/test_dates.py @@ -99,10 +99,10 @@ class DatetimeTests(unittest.TestCase, CommonDatetimeTestsMixin): def setUp(self): self.conn = psycopg2.connect(dsn) self.curs = self.conn.cursor() - self.DATE = psycopg2._psycopg.PYDATE - self.TIME = psycopg2._psycopg.PYTIME - self.DATETIME = psycopg2._psycopg.PYDATETIME - self.INTERVAL = psycopg2._psycopg.PYINTERVAL + self.DATE = psycopg2.extensions.PYDATE + self.TIME = psycopg2.extensions.PYTIME + self.DATETIME = psycopg2.extensions.PYDATETIME + self.INTERVAL = psycopg2.extensions.PYINTERVAL def tearDown(self): self.conn.close() @@ -307,7 +307,7 @@ class DatetimeTests(unittest.TestCase, CommonDatetimeTestsMixin): # Only run the datetime tests if psycopg was compiled with support. -if not hasattr(psycopg2._psycopg, 'PYDATETIME'): +if not hasattr(psycopg2.extensions, 'PYDATETIME'): del DatetimeTests @@ -484,7 +484,10 @@ class mxDateTimeTests(unittest.TestCase, CommonDatetimeTestsMixin): # Only run the mx.DateTime tests if psycopg was compiled with support. -if not hasattr(psycopg2._psycopg, 'MXDATETIME'): +try: + if not hasattr(psycopg2._psycopg, 'MXDATETIME'): + del mxDateTimeTests +except AttributeError: del mxDateTimeTests