Flake8 fixes

aviau · aviau · commit 709131a620ca · 2015-02-17T00:09:00.000-05:00
diff --git a/influxdb/dataframe_client.py b/influxdb/dataframe_client.py
@@ -46,8 +46,8 @@ def write_points(self, data, *args, **kwargs):
         if batch_size:
             kwargs.pop('batch_size')  # don't hand over to InfluxDBClient
             for key, data_frame in data.items():
-                number_batches = int(math.ceil(len(data_frame)
-                                     / float(batch_size)))
+                number_batches = int(math.ceil(
+                    len(data_frame) / float(batch_size)))
                 for batch in range(number_batches):
                     start_index = batch * batch_size
                     end_index = (batch + 1) * batch_size
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -103,33 +103,32 @@ def test_switch_user(self):
         assert cli._password == 'another_password'
 
     def test_write(self):
-          with requests_mock.Mocker() as m:
-              m.register_uri(
-                  requests_mock.POST,
-                  "http://localhost:8086/write"
-              )
-
-              cli = InfluxDBClient(database='db')
-              cli.write(
-                  {"database": "mydb",
-                   "retentionPolicy": "mypolicy",
-                   "points": [{"name": "cpu_load_short",
-                               "tags": {"host": "server01",
-                                        "region": "us-west"},
-                              "timestamp": "2009-11-10T23:00:00Z",
-                              "values": {"value": 0.64}}]}
-              )
-
-              self.assertEqual(
-                  json.loads(m.last_request.body),
-                  {"database": "mydb",
-                   "retentionPolicy": "mypolicy",
-                   "points": [{"name": "cpu_load_short",
-                               "tags": {"host": "server01",
-                                        "region": "us-west"},
-                              "timestamp": "2009-11-10T23:00:00Z",
-                              "values": {"value": 0.64}}]}
-              )
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.POST,
+                "http://localhost:8086/write"
+            )
+            cli = InfluxDBClient(database='db')
+            cli.write(
+                {"database": "mydb",
+                 "retentionPolicy": "mypolicy",
+                 "points": [{"name": "cpu_load_short",
+                             "tags": {"host": "server01",
+                                      "region": "us-west"},
+                             "timestamp": "2009-11-10T23:00:00Z",
+                             "values": {"value": 0.64}}]}
+            )
+
+            self.assertEqual(
+                json.loads(m.last_request.body),
+                {"database": "mydb",
+                 "retentionPolicy": "mypolicy",
+                 "points": [{"name": "cpu_load_short",
+                             "tags": {"host": "server01",
+                                      "region": "us-west"},
+                             "timestamp": "2009-11-10T23:00:00Z",
+                             "values": {"value": 0.64}}]}
+            )
 
     def test_write_points(self):
         with requests_mock.Mocker() as m:
@@ -539,7 +538,7 @@ def test_get_database_users(self):
         cli = InfluxDBClient('localhost', 8086, 'username', 'password', 'db')
 
         example_response = \
-            '[{"name":"paul","isAdmin":false,"writeTo":".*","readFrom":".*"},' \
+            '[{"name":"paul","isAdmin":false,"writeTo":".*","readFrom":".*"},'\
             '{"name":"bobby","isAdmin":false,"writeTo":".*","readFrom":".*"}]'
 
         with requests_mock.Mocker() as m:
diff --git a/tests/influxdb/dataframe_client_test.py b/tests/influxdb/dataframe_client_test.py
@@ -2,6 +2,8 @@
 """
 unit tests for misc module
 """
+from .client_test import _mocked_session
+
 import unittest
 import json
 import requests_mock
@@ -16,8 +18,6 @@
     from pandas.util.testing import assert_frame_equal
     from influxdb import DataFrameClient
 
-from .client_test import _mocked_session
-
 
 @skipIfPYpy
 class TestDataFrameClient(unittest.TestCase):
