make tests pass: tz_localize/tz_convert

Author: clslgrnc

influxdb/_dataframe_client.py

@@ -208,7 +208,10 @@ def _to_dataframe(self, rs, dropna=True):
             df = pd.DataFrame(data)
             df.time = pd.to_datetime(df.time)
             df.set_index('time', inplace=True)
-            df.index = df.index.tz_localize('UTC')
+            try:
+                df.index = df.index.tz_localize('UTC')
+            except TypeError:
+                df.index = df.index.tz_convert('UTC')
             df.index.name = None
             result[key].append(df)
         for key, data in result.items():

influxdb/tests/dataframe_client_test.py

@@ -818,13 +818,19 @@ def test_query_into_dataframe(self):
         pd1 = pd.DataFrame(
             [[23422]], columns=['value'],
             index=pd.to_datetime(["2009-11-10T23:00:00Z"]))
-        pd1.index = pd1.index.tz_localize('UTC')
+        try:
+            pd1.index = pd1.index.tz_localize('UTC')
+        except TypeError:
+            pd1.index = pd1.index.tz_convert('UTC')
         pd2 = pd.DataFrame(
             [[23422], [23422], [23422]], columns=['value'],
             index=pd.to_datetime(["2009-11-10T23:00:00Z",
                                   "2009-11-10T23:00:00Z",
                                   "2009-11-10T23:00:00Z"]))
-        pd2.index = pd2.index.tz_localize('UTC')
+        try:
+            pd2.index = pd2.index.tz_localize('UTC')
+        except TypeError:
+            pd2.index = pd2.index.tz_convert('UTC')
         expected = {
             ('network', (('direction', ''),)): pd1,
             ('network', (('direction', 'in'),)): pd2
@@ -871,11 +877,18 @@ def test_multiquery_into_dataframe(self):
             index=pd.to_datetime([
                 "2015-01-29 21:55:43.702900257+0000",
                 "2015-01-29 21:55:43.702900257+0000",
-                "2015-06-11 20:46:02+0000"])).tz_localize('UTC')
+                "2015-06-11 20:46:02+0000"]))
+        try:
+            pd1 = pd1.tz_localize('UTC')
+        except TypeError:
+            pd1 = pd1.tz_convert('UTC')
         pd2 = pd.DataFrame(
             [[3]], columns=['count'],
-            index=pd.to_datetime(["1970-01-01 00:00:00+00:00"]))\
-            .tz_localize('UTC')
+            index=pd.to_datetime(["1970-01-01 00:00:00+00:00"]))
+        try:
+            pd2 = pd2.tz_localize('UTC')
+        except TypeError:
+            pd2 = pd2.tz_convert('UTC')
         expected = [{'cpu_load_short': pd1}, {'cpu_load_short': pd2}]
 
         cli = DataFrameClient('host', 8086, 'username', 'password', 'db')