Add support for custom indexes for query in the DataFrameClient (infl…

…uxdata#785)
rolincova · Jun 1, 2020 · ddd8261 · ddd8261
1 parent 055d71f
commit ddd8261
Showing 1 changed file with 4 additions and 4 deletions.
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))