Fix flake8

Author: k-raina

tests/test_Admin.py

@@ -1530,86 +1530,85 @@ class UninitializedAdmin(AdminClient):
         def __init__(self, config):
             # Don't call super().__init__() - leaves self->rk as NULL
             pass
-    
+
     admin = UninitializedAdmin({})
-    
-   
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.create_topics([NewTopic("test", 1, 1)])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.delete_topics(["test"])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.create_partitions([NewPartitions("test", 2)])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_configs([ConfigResource(ResourceType.TOPIC, "test")])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.incremental_alter_configs([ConfigResource(ResourceType.TOPIC, "test")])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.alter_configs([ConfigResource(ResourceType.TOPIC, "test")])
-    
+
     acl_binding = AclBinding(
         ResourceType.TOPIC, "topic1", ResourcePatternType.LITERAL,
         "User:u1", "*", AclOperation.WRITE, AclPermissionType.ALLOW
     )
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.create_acls([acl_binding])
-    
+
     acl_filter = AclBindingFilter(ResourceType.ANY, None, ResourcePatternType.ANY,
-                                   None, None, AclOperation.ANY, AclPermissionType.ANY)
-    
+                                  None, None, AclOperation.ANY, AclPermissionType.ANY)
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.delete_acls([acl_filter])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_acls(acl_filter)
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.list_consumer_groups()
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_user_scram_credentials(["user"])
-    
+
     scram_info = ScramCredentialInfo(ScramMechanism.SCRAM_SHA_256, 10000)
     upsertion = UserScramCredentialUpsertion("user", scram_info, b"password")
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.alter_user_scram_credentials([upsertion])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_consumer_groups(["group"])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_topics(TopicCollection(["topic"]))
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.describe_cluster()
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.delete_consumer_groups(["group"])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.list_consumer_group_offsets([ConsumerGroupTopicPartitions("group")])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.alter_consumer_group_offsets([ConsumerGroupTopicPartitions("group", [TopicPartition("topic", 0, 5)])])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.list_offsets({TopicPartition("topic", 0, 10): OffsetSpec.earliest()})
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.delete_records([TopicPartition("topic", 0, 10)])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.elect_leaders(ElectionType.PREFERRED, [TopicPartition("topic", 0)])
-    
+
     with pytest.raises(RuntimeError, match="AdminClient has been closed"):
         admin.poll(0.001)
-    
+
     # Test __len__() - should return 0 for closed admin (safe, no crash)
     assert len(admin) == 0

tests/test_Consumer.py

@@ -624,56 +624,55 @@ class UninitializedConsumer(Consumer):
         def __init__(self, config):
             # Don't call super().__init__() - leaves self->rk as NULL
             pass
-    
+
     consumer = UninitializedConsumer({'group.id': 'test'})
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.subscribe(['topic'])
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.unsubscribe()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.poll()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.consume()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.assign([TopicPartition('topic', 0)])
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.unassign()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.assignment()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.commit()
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.committed([TopicPartition('topic', 0)])
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.position([TopicPartition('topic', 0)])
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.seek(TopicPartition('topic', 0, 0))
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.get_watermark_offsets(TopicPartition('topic', 0))
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.store_offsets([TopicPartition('topic', 0, 42)])
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.offsets_for_times([TopicPartition('topic', 0)])
-    
+
     with pytest.raises(RuntimeError, match="Handle has been closed"):
         consumer.list_topics()
-    
 
     consumer.close()  # Should succeed
-    
+
     with pytest.raises(RuntimeError, match="Consumer closed"):
         consumer.consumer_group_metadata()

tests/test_Producer.py

@@ -1347,51 +1347,51 @@ def on_delivery(err, msg):
 
 def test_uninitialized_producer_methods():
     """Test that all Producer methods raise RuntimeError when called on uninitialized instance.
-    
+
     This test verifies issue #1590 fix - prevents SEGV when subclassing Producer
     without calling super().__init__().
     """
     class UninitializedProducer(Producer):
         def __init__(self, config):
             # Don't call super().__init__() - leaves self->rk as NULL
             pass
-    
+
     producer = UninitializedProducer({})
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.produce('topic', value=b'test')
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.poll()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.flush()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.purge()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.produce_batch('topic', [{'value': b'test'}])
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.init_transactions()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.begin_transaction()
-    
+
     # send_offsets_to_transaction - NULL check happens after argument parsing
     consumer = Consumer({'group.id': 'test', 'socket.timeout.ms': 10})
     metadata = consumer.consumer_group_metadata()
     consumer.close()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.send_offsets_to_transaction([TopicPartition('topic', 0)], metadata)
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.commit_transaction()
-    
+
     with pytest.raises(RuntimeError, match="Producer has been closed"):
         producer.abort_transaction()
-    
+
     # Test __len__() - should return 0 for closed producer (safe, no crash)
     assert len(producer) == 0