Fix flake8 issues

rnpridgeon · rnpridgeon · commit 0fb364a77417 · 2019-01-04T20:53:57.000-05:00
diff --git a/confluent_kafka/avro/schema.py b/confluent_kafka/avro/schema.py
@@ -26,18 +26,18 @@
 
 class GenericAvroRecord(dict):
     """
-    Pairs a AvroRecord with it's schema
+    Pairs an AvroRecord with it's schema
 
     :param schema schema: A parsed Avro schema.
-    :param dict record: Wraps existing dict in GenericAvroRecord
-    :raises ValueError: If schema is None
-    :returns: Avro record with its schema 
+    :param dict record: Wraps existing dict in GenericAvroRecord.
+    :raises ValueError: If schema is None.
+    :returns: Avro record with its schema.
     :rtype: GenericAvroRecord
     """
     __slots__ = ['schema']
 
     def __init__(self, schema, record=None):
-        if schema is None: 
+        if schema is None:
             raise ValueError("schema must not be None")
         self.schema = schema
         if record is not None:
@@ -61,4 +61,4 @@ def get_schema(datum):
     elif isinstance(datum, float):
         return PrimitiveSchema('float')
     else:
-        raise ValueError("Unsupported Avro type {}.".format(type(datum)))
+        raise ValueError("Unsupported Avro type {}.".format(type(datum)))
diff --git a/confluent_kafka/producer.py b/confluent_kafka/producer.py
@@ -139,5 +139,6 @@ def produce(self, topic, value=None, key=None, partition=PARTITION_UA,
 
         """
 
-        super(SerializingProducer, self).produce(topic, self._value_serializer(topic, value), self._key_serializer(topic, key), partition, on_delivery=on_delivery,
+        super(SerializingProducer, self).produce(topic, self._value_serializer(topic, value),
+                                                 self._key_serializer(topic, key), partition, on_delivery=on_delivery,
                                                  callback=callback, timestamp=timestamp, headers=headers)
diff --git a/tests/avro/test_avro_serializer.py b/tests/avro/test_avro_serializer.py
@@ -81,4 +81,3 @@ def test_decode_none(self):
         """"null/None messages should decode to None"""
 
         self.assertIsNone(self.deserializer(None, None))
-
diff --git a/tests/integration/integration_test.py b/tests/integration/integration_test.py
@@ -1232,7 +1232,10 @@ def verify_serializer():
     from base64 import standard_b64encode as b64encode
     from base64 import standard_b64decode as b64decode
 
-    from confluent_kafka.avro.serializer import AvroSerializer, AvroDeserializer, RecordNameStrategy, TopicRecordNameStrategy
+    from confluent_kafka.avro.serializer import (AvroSerializer,
+                                                 AvroDeserializer,
+                                                 RecordNameStrategy,
+                                                 TopicRecordNameStrategy)
     from confluent_kafka.avro.schema import GenericAvroRecord
     from confluent_kafka import avro
 
@@ -1261,7 +1264,7 @@ def b64_decoder(topic, data):
              key_deserializer=AvroDeserializer(sr, is_key=True)),
         dict(key=b'string serializer', value=GenericAvroRecord(schema, {'name': 'avro serializer', 'number': 2}),
              value_serializer=AvroSerializer(sr, is_key=False,
-             value_deserializer=AvroDeserializer(sr, is_key=False))),
+                                             value_deserializer=AvroDeserializer(sr, is_key=False))),
         dict(key=b'base64 encoder', value=GenericAvroRecord(schema, {'name': 'avro serializer', 'number': 3}),
              key_serializer=lambda topic, key: b64encode(key),
              value_serializer=AvroSerializer(sr, subject_strategy=TopicRecordNameStrategy),

Original file line number	Diff line number	Diff line change
`@@ -81,4 +81,3 @@ def test_decode_none(self):`
`81`	`81`	`""""null/None messages should decode to None"""`
`82`	`82`
`83`	`83`	`self.assertIsNone(self.deserializer(None, None))`
`84`		`-`