skip tests if mx.DateTime is not installed

1 files changed, 8 insertions, 1 deletions

diff --git a/test/unittest_date.py b/test/unittest_date.py
index 55c03d6..fdcb644 100644
--- a/test/unittest_date.py
+++ b/test/unittest_date.py
@@ -5,12 +5,17 @@ Unittests for date helpers
 from logilab.common.testlib import TestCase, unittest_main
 from logilab.common.date import date_range, endOfMonth
 
-from mx.DateTime import Date, RelativeDate
+try:
+    from mx.DateTime import Date, RelativeDate
+except ImportError:
+    Date = None
 
 class DateTC(TestCase):
 
     def test_day(self):
         """enumerate days"""
+        if Date is None:
+            self.skip('mx.DateTime is not installed')
         r = list(date_range(Date(2000,1,1), Date(2000,1,4)))
         expected = [Date(2000,1,1), Date(2000,1,2), Date(2000,1,3)]
         self.assertListEquals(r, expected)
@@ -20,6 +25,8 @@ class DateTC(TestCase):
 
     def test_month(self):
         """enumerate months"""
+        if Date is None:
+            self.skip('mx.DateTime is not installed')
         r = list(date_range(Date(2000,1,2), Date(2000,4,4), endOfMonth))
         expected = [Date(2000,1,2), Date(2000,2,29), Date(2000,3,31)]
         self.assertListEquals(r, expected)