SNOW-67593: Added numpy bool_ support

Author: smtakeda

compat.py

@@ -21,7 +21,7 @@
 
     NUM_DATA_TYPES = [numpy.int8, numpy.int16, numpy.int32, numpy.int64,
                       numpy.float16, numpy.float32, numpy.float64,
-                      numpy.uint8, numpy.uint16, numpy.uint32, numpy.uint64]
+                      numpy.uint8, numpy.uint16, numpy.uint32, numpy.uint64, numpy.bool_]
 except:
     numpy = None
 

converter.py

@@ -49,6 +49,7 @@
     u'bytes': u'BINARY',
     u'bytearray': u'BINARY',
     u'bool': u'BOOLEAN',
+    u'bool_': u'BOOLEAN',
     u'nonetype': u'ANY',
     u'datetime': u'TIMESTAMP_NTZ',
     u'sfdatetime': u'TIMESTAMP_NTZ',
@@ -519,6 +520,9 @@ def _bytes_to_snowflake(self, value):
     def _bool_to_snowflake(self, value):
         return value
 
+    def _bool__to_snowflake(self, value):
+        return bool(value)
+
     def _nonetype_to_snowflake(self, _):
         return None
 

test/test_numpy_binding.py

@@ -13,27 +13,31 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
     all_data = [{
         'tz': 'America/Los_Angeles',
         'float': '1.79769313486e+308',
+        'numpy_bool': np.True_,
         'epoch_time': epoch_time,
         'current_time': current_datetime64,
         'specific_date': np.datetime64('2005-02-25T03:30'),
         'expected_specific_date': np.datetime64('2005-02-25T03:30').astype(datetime.datetime)
     }, {
         'tz': 'Asia/Tokyo',
         'float': '-1.79769313486e+308',
+        'numpy_bool': np.False_,
         'epoch_time': epoch_time,
         'current_time': current_datetime64,
         'specific_date': np.datetime64('1970-12-31T05:00:00'),
         'expected_specific_date': np.datetime64('1970-12-31T05:00:00').astype(datetime.datetime)
     }, {
         'tz': 'America/New_York',
         'float': '-1.79769313486e+308',
+        'numpy_bool': np.True_,
         'epoch_time': epoch_time,
         'current_time': current_datetime64,
         'specific_date': np.datetime64('1969-12-31T05:00:00'),
         'expected_specific_date': np.datetime64('1969-12-31T05:00:00').astype(datetime.datetime)
     }, {
         'tz': 'UTC',
         'float': '-1.79769313486e+308',
+        'numpy_bool': np.False_,
         'epoch_time': epoch_time,
         'current_time': current_datetime64,
         'specific_date': np.datetime64('1968-11-12T07:00:00.123'),
@@ -53,7 +57,8 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
     c8  timestamp_ntz, -- datetime64
     c9  date,          -- datetime64
     c10 timestamp_ltz, -- datetime64,
-    c11 timestamp_tz)  -- datetime64
+    c11 timestamp_tz,  -- datetime64
+    c12 boolean)       -- numpy.bool_
             """.format(name=db_parameters['name']))
             for data in all_data:
                 cnx.cursor().execute("""
@@ -70,7 +75,8 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
     c8,
     c9,
     c10,
-    c11
+    c11,
+    c12
 )
 VALUES(
     %s,
@@ -83,6 +89,7 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
     %s,
     %s,
     %s,
+    %s,
     %s)""".format(
                     name=db_parameters['name']), (
                     np.iinfo(np.int8).max,
@@ -96,6 +103,7 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
                     data['current_time'],
                     data['current_time'],
                     data['specific_date'],
+                    data['numpy_bool']
                 ))
                 rec = cnx.cursor().execute("""
 SELECT
@@ -109,7 +117,8 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
        c8,
        c9,
        c10,
-       c11
+       c11,
+       c12
   FROM {name}""".format(
                     name=db_parameters['name'])).fetchone()
                 assert np.int8(rec[0]) == np.iinfo(np.int8).max
@@ -123,6 +132,7 @@ def test_numpy_datatype_binding(conn_cnx, db_parameters):
                 assert str(rec[8]) == str(data['current_time'])[0:10]
                 assert rec[9] == datetime.datetime.fromtimestamp(epoch_time, rec[9].tzinfo)
                 assert rec[10] == data['expected_specific_date'].replace(tzinfo=rec[10].tzinfo)
+                assert isinstance(rec[11], bool) and rec[11] == data['numpy_bool'] and np.bool_(rec[11]) == data['numpy_bool']
                 cnx.cursor().execute("""
 DELETE FROM {name}""".format(name=db_parameters['name']))
     finally: