unit test for dateutil and pytz

tests/types/test_timezone.py

@@ -4,21 +4,18 @@ import sqlalchemy as sa
 from sqlalchemy_utils.types import timezone
 from tests import TestCase
 
-
-try:
-    import dateutil
-
-except ImportError:
-    dateutil = None
+import dateutil
+import pytz
 
 
-@mark.skipif('dateutil is None')
 class TestTimezoneType(TestCase):
     def create_models(self):
         class Visitor(self.Base):
             __tablename__ = 'document'
             id = sa.Column(sa.Integer, primary_key=True)
-            timezone = sa.Column(timezone.TimezoneType)
+            timezone_dateutil = sa.Column(timezone.TimezoneType(
+                                          backend='dateutil'))
+            timezone_pytz = sa.Column(timezone.TimezoneType(backend='pytz'))
 
             def __repr__(self):
                 return 'Visitor(%r)' % self.id
@@ -27,13 +24,17 @@ class TestTimezoneType(TestCase):
 
     def test_parameter_processing(self):
         visitor = self.Visitor(
-            timezone=u'America/Los_Angeles'
+            timezone_dateutil=u'America/Los_Angeles',
+            timezone_pytz=u'America/Los_Angeles'
         )
 
         self.session.add(visitor)
         self.session.commit()
 
-        visitor = self.session.query(self.Visitor).filter_by(
-            timezone='America/Los_Angeles').first()
+        visitor_dateutil = self.session.query(self.Visitor).filter_by(
+            timezone_dateutil='America/Los_Angeles').first()
+        visitor_pytz = self.session.query(self.Visitor).filter_by(
+            timezone_pytz='America/Los_Angeles').first()
 
-        assert visitor is not None
+        assert visitor_dateutil is not None
+        assert visitor_pytz is not None