Add support for custom indexes for query in the DataFrameClient (#785)

rolincova · rolincova · commit ddd82612f57d · 2020-06-01T10:40:39.000+02:00
diff --git a/influxdb/tests/dataframe_client_test.py b/influxdb/tests/dataframe_client_test.py
@@ -1266,11 +1266,11 @@ def test_query_custom_index(self):
               "SELECT count(value) FROM cpu_load_short WHERE region=$region"
         bind_params = {'region': 'us-west'}
         with _mocked_session(cli, 'GET', 200, data):
-            result = cli.query(iql, bind_params=bind_params, data_frame_index=["time", "host"])
+            result = cli.query(iql, bind_params=bind_params,
+                               data_frame_index=["time", "host"])
 
             _data_frame = result['cpu_load_short']
             print(_data_frame)
 
-            self.assertListEqual(["time", "host"], list(_data_frame.index.names))
-
-
+            self.assertListEqual(["time", "host"],
+                                 list(_data_frame.index.names))
