Fixed the error 'module' object is not callable in csvsql when loading Snowflake SQLAlchemy dialect

Author: smtakeda

setup.py

@@ -50,7 +50,7 @@
     },
     entry_points={
         'sqlalchemy.dialects': [
-            'snowflake=snowflake.sqlalchemy.base',
+            'snowflake=snowflake.sqlalchemy.base:SnowflakeDialect',
         ]
     },
     classifiers = [

test/test_core.py

@@ -12,10 +12,12 @@
                         Sequence, ForeignKey, Binary, REAL)
 from sqlalchemy import inspect
 from sqlalchemy import text
+from sqlalchemy import dialects
 from sqlalchemy.sql import and_, or_, not_
 from sqlalchemy.sql import select
 
 from snowflake.sqlalchemy import URL
+from snowflake.sqlalchemy.base import SnowflakeDialect
 
 try:
     from parameters import (CONNECTION_PARAMETERS2)
@@ -794,3 +796,10 @@ def test_azure():
         assert 'Failed to connect to DB' in ex.orig.msg
         assert 'testaccount.east-us-2.azure.snowflakecomputing.com' in \
                ex.orig.msg
+
+
+def test_load_dialect():
+    """
+    Test loading Snowflake SQLAlchemy dialect class
+    """
+    assert isinstance(dialects.registry.load('snowflake')(), SnowflakeDialect)

test/test_pandas.py

@@ -111,7 +111,7 @@ def get_engine_with_numpy(db_parameters, user=None, password=None,
 def test_numpy_datatypes(db_parameters):
     engine = get_engine_with_numpy(db_parameters)
     try:
-        specific_date = np.datetime64('2016-03-04T12:03:05.123456789Z')
+        specific_date = np.datetime64('2016-03-04T12:03:05.123456789')
         engine.execute(
             "CREATE OR REPLACE TABLE {name}("
             "c1 timestamp_ntz)".format(name=db_parameters['name']))