fix: serialize numpy floats (influxdata#254)

bednar · web-flow · commit a421a6f54269 · 2021-05-19T13:41:24.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,9 @@
 1. [#241](https://github.com/influxdata/influxdb-client-python/pull/241): Add detail error message for not supported type of `Point.field`
 1. [#238](https://github.com/influxdata/influxdb-client-python/pull/238): Add possibility to specify default `timezone` for datetimes without `tzinfo`
 
+### Bug Fixes
+1. [#254](https://github.com/influxdata/influxdb-client-python/pull/254): Serialize `numpy` floats into LineProtocol
+
 ### Documentation
 1. [#255](https://github.com/influxdata/influxdb-client-python/pull/255): Fix invalid description for env var `INFLUXDB_V2_CONNECTION_POOL_MAXSIZE`
 
diff --git a/influxdb_client/client/write/dataframe_serializer.py b/influxdb_client/client/write/dataframe_serializer.py
@@ -158,7 +158,7 @@ def data_frame_to_list_of_points(data_frame, point_settings, **kwargs):
             field_value = f"{sep}{key_format}={{{val_format}}}i"
         elif issubclass(value.type, np.bool_):
             field_value = f'{sep}{key_format}={{{val_format}}}'
-        elif issubclass(value.type, np.float):
+        elif issubclass(value.type, np.floating):
             if null_columns[index]:
                 field_value = f"""{{"" if math.isnan({val_format}) else f"{sep}{key_format}={{{val_format}}}"}}"""
             else:
diff --git a/tests/test_WriteApiDataFrame.py b/tests/test_WriteApiDataFrame.py
@@ -1,5 +1,3 @@
-import csv
-import os
 import time
 import unittest
 from datetime import timedelta
@@ -160,7 +158,7 @@ def test_write_object_field_nan(self):
                          points[1])
 
     def test_write_field_bool(self):
-        from influxdb_client.extras import pd, np
+        from influxdb_client.extras import pd
 
         now = pd.Timestamp('2020-04-05 00:00+00:00')
 
@@ -252,7 +250,7 @@ def test_tags_order(self):
         self.assertEqual("h2o,a=a,b=b,c=c level=2i 1586048400000000000", points[0])
 
     def test_escape_text_value(self):
-        from influxdb_client.extras import pd, np
+        from influxdb_client.extras import pd
 
         now = pd.Timestamp('2020-04-05 00:00+00:00')
         an_hour_ago = now - timedelta(hours=1)
@@ -273,7 +271,7 @@ def test_escape_text_value(self):
         self.assertEqual("test,d=bar\\ foo b=\"goodbye cruel world\",c=2i 1586044800000000000", points[1])
 
     def test_with_default_tags(self):
-        from influxdb_client.extras import pd, np
+        from influxdb_client.extras import pd
         now = pd.Timestamp('2020-04-05 00:00+00:00')
         data_frame = pd.DataFrame(data={
                                       'value': [1, 2],
@@ -309,8 +307,7 @@ def test_with_default_tags(self):
         self.assertEqual(True, (data_frame == original_data).all(axis = None), f'data changed; old:\n{original_data}\nnew:\n{data_frame}')
 
     def test_with_period_index(self):
-        from influxdb_client.extras import pd, np
-        now = pd.Timestamp('2020-04-05 00:00+00:00')
+        from influxdb_client.extras import pd
         data_frame = pd.DataFrame(data={
                                       'value': [1, 2],
                                 },
@@ -323,3 +320,16 @@ def test_with_period_index(self):
         self.assertEqual(2, len(points))
         self.assertEqual("h2o value=1i 1586048400000000000", points[0])
         self.assertEqual("h2o value=2i 1586052000000000000", points[1])
+
+    def test_write_num_py_floats(self):
+        from influxdb_client.extras import pd, np
+        now = pd.Timestamp('2020-04-05 00:00+00:00')
+
+        for np_float_type in [np.float, np.float16, np.float32, np.float64, np.float128]:
+            data_frame = pd.DataFrame([15.5], index=[now], columns=['level']).astype(np_float_type)
+            points = data_frame_to_list_of_points(data_frame=data_frame,
+                                                  data_frame_measurement_name='h2o',
+                                                  point_settings=PointSettings())
+            self.assertEqual(1, len(points))
+            self.assertEqual("h2o level=15.5 1586044800000000000", points[0], msg=f'Current type: {np_float_type}')
+
