Fix linting errors on test files

Author: pberganza-applaudostudios

analytics/test/__init__.py

@@ -3,10 +3,12 @@
 import logging
 import sys
 
+
 def all_names():
     for _, modname, _ in pkgutil.iter_modules(__path__):
         yield 'analytics.test.' + modname
 
+
 def all():
     logging.basicConfig(stream=sys.stderr)
     return unittest.defaultTestLoader.loadTestsFromNames(all_names())

analytics/test/client.py

@@ -326,7 +326,8 @@ def mock_post_fn(*args, **kwargs):
 
         # the post function should be called 2 times, with a batch size of 10
         # each time.
-        with mock.patch('analytics.consumer.post', side_effect=mock_post_fn) as mock_post:
+        with mock.patch('analytics.consumer.post', side_effect=mock_post_fn) \
+                as mock_post:
             for _ in range(20):
                 client.identify('userId', {'trait': 'value'})
             time.sleep(1)

analytics/test/consumer.py

@@ -52,8 +52,9 @@ def test_upload(self):
         self.assertTrue(success)
 
     def test_flush_interval(self):
-        # Put _n_ items in the queue, pausing a little bit more than _flush_interval_
-        # after each one. The consumer should upload _n_ times.
+        # Put _n_ items in the queue, pausing a little bit more than
+        # _flush_interval_ after each one.
+        # The consumer should upload _n_ times.
         q = Queue()
         flush_interval = 0.3
         consumer = Consumer(q, 'testsecret', flush_at=10,
@@ -71,8 +72,8 @@ def test_flush_interval(self):
             self.assertEqual(mock_post.call_count, 3)
 
     def test_multiple_uploads_per_interval(self):
-        # Put _flush_at*2_ items in the queue at once, then pause for _flush_interval_.
-        # The consumer should upload 2 times.
+        # Put _flush_at*2_ items in the queue at once, then pause for
+        # _flush_interval_. The consumer should upload 2 times.
         q = Queue()
         flush_interval = 0.5
         flush_at = 10
@@ -99,34 +100,38 @@ def test_request(self):
         }
         consumer.request([track])
 
-    def _test_request_retry(self, consumer, expected_exception, exception_count):
+    def _test_request_retry(self, consumer,
+                            expected_exception, exception_count):
 
         def mock_post(*args, **kwargs):
             mock_post.call_count += 1
             if mock_post.call_count <= exception_count:
                 raise expected_exception
         mock_post.call_count = 0
 
-        with mock.patch('analytics.consumer.post', mock.Mock(side_effect=mock_post)):
+        with mock.patch('analytics.consumer.post',
+                        mock.Mock(side_effect=mock_post)):
             track = {
                 'type': 'track',
                 'event': 'python event',
                 'userId': 'userId'
             }
-            # request() should succeed if the number of exceptions raised is less
-            # than the retries paramater.
+            # request() should succeed if the number of exceptions raised is
+            # less than the retries paramater.
             if exception_count <= consumer.retries:
                 consumer.request([track])
             else:
-                # if exceptions are raised more times than the retries parameter,
-                # we expect the exception to be returned to the caller.
+                # if exceptions are raised more times than the retries
+                # parameter, we expect the exception to be returned to
+                # the caller.
                 try:
                     consumer.request([track])
                 except type(expected_exception) as exc:
                     self.assertEqual(exc, expected_exception)
                 else:
                     self.fail(
-                        "request() should raise an exception if still failing after %d retries" % consumer.retries)
+                        "request() should raise an exception if still failing "
+                        "after %d retries" % consumer.retries)
 
     def test_request_retry(self):
         # we should retry on general errors
@@ -180,10 +185,12 @@ def mock_post_fn(_, data, **kwargs):
             res = mock.Mock()
             res.status_code = 200
             self.assertTrue(len(data.encode()) < 500000,
-                            'batch size (%d) exceeds 500KB limit' % len(data.encode()))
+                            'batch size (%d) exceeds 500KB limit'
+                            % len(data.encode()))
             return res
 
-        with mock.patch('analytics.request._session.post', side_effect=mock_post_fn) as mock_post:
+        with mock.patch('analytics.request._session.post',
+                        side_effect=mock_post_fn) as mock_post:
             consumer.start()
             for _ in range(0, n_msgs + 2):
                 q.put(track)

analytics/test/module.py

@@ -26,7 +26,7 @@ def test_track(self):
         analytics.flush()
 
     def test_identify(self):
-        analytics.identify('userId', { 'email': '[email protected]' })
+        analytics.identify('userId', {'email': '[email protected]'})
         analytics.flush()
 
     def test_group(self):

analytics/test/request.py

@@ -17,13 +17,15 @@ def test_valid_request(self):
         self.assertEqual(res.status_code, 200)
 
     def test_invalid_request_error(self):
-        self.assertRaises(Exception, post, 'testsecret', 'https://api.segment.io', False, '[{]')
+        self.assertRaises(Exception, post, 'testsecret',
+                          'https://api.segment.io', False, '[{]')
 
     def test_invalid_host(self):
-        self.assertRaises(Exception, post, 'testsecret', 'api.segment.io/', batch=[])
+        self.assertRaises(Exception, post, 'testsecret',
+                          'api.segment.io/', batch=[])
 
     def test_datetime_serialization(self):
-        data = { 'created': datetime(2012, 3, 4, 5, 6, 7, 891011) }
+        data = {'created': datetime(2012, 3, 4, 5, 6, 7, 891011)}
         result = json.dumps(data, cls=DatetimeSerializer)
         self.assertEqual(result, '{"created": "2012-03-04T05:06:07.891011"}')
 
@@ -44,7 +46,7 @@ def test_should_not_timeout(self):
 
     def test_should_timeout(self):
         with self.assertRaises(requests.ReadTimeout):
-            res = post('testsecret', batch=[{
+            post('testsecret', batch=[{
                 'userId': 'userId',
                 'event': 'python event',
                 'type': 'track'

analytics/test/utils.py

@@ -65,12 +65,14 @@ def test_bytes(self):
         utils.clean(item)
 
     def test_clean_fn(self):
-        cleaned = utils.clean({ 'fn': lambda x: x, 'number': 4 })
+        cleaned = utils.clean({'fn': lambda x: x, 'number': 4})
         self.assertEqual(cleaned['number'], 4)
         # TODO: fixme, different behavior on python 2 and 3
         if 'fn' in cleaned:
             self.assertEqual(cleaned['fn'], None)
 
     def test_remove_slash(self):
-        self.assertEqual('http://segment.io', utils.remove_trailing_slash('http://segment.io/'))
-        self.assertEqual('http://segment.io', utils.remove_trailing_slash('http://segment.io'))
+        self.assertEqual('http://segment.io',
+                         utils.remove_trailing_slash('http://segment.io/'))
+        self.assertEqual('http://segment.io',
+                         utils.remove_trailing_slash('http://segment.io'))