Style fixes

Author: pranavrth

tests/common/__init__.py

@@ -60,11 +60,14 @@ def update_conf_group_protocol(conf=None):
 
     @staticmethod
     def can_upgrade_group_protocol_to_consumer(conf):
-        return conf is not None and 'group.id' in conf and 'group.protocol' not in conf and TestUtils.use_group_protocol_consumer()
+        return (conf is not None and 'group.id' in conf and
+                'group.protocol' not in conf and TestUtils.use_group_protocol_consumer())
 
     @staticmethod
     def remove_forbidden_conf_group_protocol_consumer(conf):
-        if conf is None or not TestUtils.use_group_protocol_consumer() or conf.get('group.protocol', 'consumer') != 'consumer':
+        if (conf is None or
+                not TestUtils.use_group_protocol_consumer() or
+                conf.get('group.protocol', 'consumer') != 'consumer'):
             return
         forbidden_conf_properties = ["session.timeout.ms",
                                      "partition.assignment.strategy",

tests/integration/cluster_fixture.py

@@ -233,7 +233,7 @@ def create_topic(self, prefix, conf=None, **create_topic_kwargs):
 
         future_topic.get(name).result()
         return name
-    
+
     def delete_topic(self, topic):
         """
         Deletes a topic with this cluster.

tests/integration/consumer/test_consumer_upgrade_downgrade.py

@@ -73,15 +73,16 @@ class Operation(Enum):
     REMOVE = 1
 
 
-def perform_consumer_upgrade_downgrade_test_with_partition_assignment_strategy(kafka_cluster, partition_assignment_strategy):
+def perform_consumer_upgrade_downgrade_test_with_partition_assignment_strategy(
+        kafka_cluster, partition_assignment_strategy):
     """
     Test consumer upgrade and downgrade.
     """
     topic_name_prefix = f"{topic_prefix}_{partition_assignment_strategy}"
     topic = kafka_cluster.create_topic_and_wait_propogation(topic_name_prefix,
-                                                                    {
-                                                                        "num_partitions": number_of_partitions
-                                                                    })
+                                                            {
+                                                                "num_partitions": number_of_partitions
+                                                            })
     admin_client = kafka_cluster.admin()
 
     consumer_conf = {'group.id': topic,