Fix linter

Author: k-raina

tests/test_Admin.py

@@ -658,20 +658,20 @@ def test_describe_topics_api():
 
         # Wrong argument type
         for args in [
-                        [topic_names],
-                        ["test-topic-1"],
-                        [TopicCollection([3])],
-                        [TopicCollection(["correct", 3])],
-                        [TopicCollection([None])]
-                    ]:
+            [topic_names],
+            ["test-topic-1"],
+            [TopicCollection([3])],
+            [TopicCollection(["correct", 3])],
+            [TopicCollection([None])]
+        ]:
             with pytest.raises(TypeError):
                 a.describe_topics(*args, **kwargs)
 
         # Wrong argument value
         for args in [
-                        [TopicCollection([""])],
-                        [TopicCollection(["correct", ""])]
-                    ]:
+            [TopicCollection([""])],
+            [TopicCollection(["correct", ""])]
+        ]:
             with pytest.raises(ValueError):
                 a.describe_topics(*args, **kwargs)
 
@@ -1053,13 +1053,13 @@ def test_list_offsets_api():
 
     # Wrong option types
     for kwargs in [
-                        {
-                            "isolation_level": 10
-                        },
-                        {
-                            "request_timeout": "test"
-                        }
-                    ]:
+        {
+            "isolation_level": 10
+        },
+        {
+            "request_timeout": "test"
+        }
+    ]:
         requests = {
             TopicPartition("topic1", 0, 10): OffsetSpec.earliest()
         }
@@ -1068,10 +1068,10 @@ def test_list_offsets_api():
 
     # Wrong option values
     for kwargs in [
-                    {
-                        "request_timeout": -1
-                    }
-                  ]:
+        {
+            "request_timeout": -1
+        }
+    ]:
         requests = {
             TopicPartition("topic1", 0, 10): OffsetSpec.earliest()
         }
@@ -1095,13 +1095,13 @@ def test_list_offsets_api():
 
         # Invalid TopicPartition
         for requests in [
-                            {
-                                TopicPartition("", 0, 10): OffsetSpec.earliest()
-                            },
-                            {
-                                TopicPartition("correct", -1, 10): OffsetSpec.earliest()
-                            }
-                        ]:
+            {
+                TopicPartition("", 0, 10): OffsetSpec.earliest()
+            },
+            {
+                TopicPartition("correct", -1, 10): OffsetSpec.earliest()
+            }
+        ]:
             with pytest.raises(ValueError):
                 a.list_offsets(requests, **kwargs)
 
@@ -1131,33 +1131,33 @@ def test_list_offsets_api():
 
         # Key isn't a TopicPartition
         for requests in [
-                            {
-                                "not-topic-partition": OffsetSpec.latest()
-                            },
-                            {
-                                TopicPartition("topic1", 0, 10): OffsetSpec.latest(),
-                                "not-topic-partition": OffsetSpec.latest()
-                            },
-                            {
-                                None: OffsetSpec.latest()
-                            }
-                        ]:
+            {
+                "not-topic-partition": OffsetSpec.latest()
+            },
+            {
+                TopicPartition("topic1", 0, 10): OffsetSpec.latest(),
+                "not-topic-partition": OffsetSpec.latest()
+            },
+            {
+                None: OffsetSpec.latest()
+            }
+        ]:
             with pytest.raises(TypeError):
                 a.list_offsets(requests, **kwargs)
 
         # Value isn't a OffsetSpec
         for requests in [
-                            {
-                                TopicPartition("topic1", 0, 10): "test"
-                            },
-                            {
-                                TopicPartition("topic1", 0, 10): OffsetSpec.latest(),
-                                TopicPartition("topic1", 0, 10): "test"
-                            },
-                            {
-                                TopicPartition("topic1", 0, 10): None
-                            }
-                        ]:
+            {
+                TopicPartition("topic1", 0, 10): "test"
+            },
+            {
+                TopicPartition("topic1", 0, 10): OffsetSpec.latest(),
+                TopicPartition("topic1", 0, 10): "test"
+            },
+            {
+                TopicPartition("topic1", 0, 10): None
+            }
+        ]:
             with pytest.raises(TypeError):
                 a.list_offsets(requests, **kwargs)
 
@@ -1226,8 +1226,6 @@ def test_elect_leaders():
             .result(timeout=1)
 
 
-@pytest.mark.skipif(libversion()[1] < 0x000b0500,
-                    reason="AdminAPI requires librdkafka >= v0.11.5")
 def test_admin_callback_exception_no_system_error():
     """Test AdminClient callbacks exception handling with different exception types"""
 
@@ -1275,10 +1273,9 @@ def error_cb_runtime_error(error):
     assert "RuntimeError from error_cb" in str(exc_info.value)
 
 
-@pytest.mark.skipif(libversion()[1] < 0x000b0500,
-                    reason="AdminAPI requires librdkafka >= v0.11.5")
 def test_admin_multiple_callbacks_different_error_types():
-    """Test AdminClient with multiple callbacks configured with different error types to see which one gets triggered"""
+    """Test AdminClient with multiple callbacks configured with different error types
+to see which one gets triggered"""
 
     callbacks_called = []
 
@@ -1304,9 +1301,9 @@ def throttle_cb_that_raises_kafka(throttle_event):
     })
 
     # Test that error_cb callback raises an exception (it's triggered by connection failures)
-    with pytest.raises(RuntimeError) as exc_info:
+    with pytest.raises(RuntimeError):
         admin.poll(timeout=0.2)
 
     # Verify that error_cb was called
     assert len(callbacks_called) > 0
-    assert 'error_cb_runtime' in callbacks_called
+    assert 'error_cb_runtime' in callbacks_called

tests/test_Consumer.py

@@ -321,7 +321,7 @@ def test_callback_exception_no_system_error():
 
     # Test error_cb callback
     error_called = []
-    
+
     def error_cb_that_raises(error):
         """Error callback that raises an exception"""
         error_called.append(error)
@@ -348,7 +348,7 @@ def error_cb_that_raises(error):
 
     # Test stats_cb callback
     stats_called = []
-    
+
     def stats_cb_that_raises(stats_json):
         """Stats callback that raises an exception"""
         stats_called.append(stats_json)
@@ -376,7 +376,7 @@ def stats_cb_that_raises(stats_json):
 
     # Test throttle_cb callback (may not be triggered in this scenario)
     throttle_called = []
-    
+
     def throttle_cb_that_raises(throttle_event):
         """Throttle callback that raises an exception"""
         throttle_called.append(throttle_event)
@@ -402,25 +402,25 @@ def throttle_cb_that_raises(throttle_event):
         if "Test exception from throttle_cb" in str(exc_info.value):
             assert len(throttle_called) > 0
             print("Throttle callback was triggered and raised exception")
-    
+
     consumer3.close()
 
 
 def test_error_callback_exception_different_error_types():
     """Test error callback with different exception types"""
-    
+
     def error_cb_kafka_exception(error):
         """Error callback that raises KafkaException"""
         raise KafkaException(error)
-    
+
     def error_cb_value_error(error):
         """Error callback that raises ValueError"""
         raise ValueError(f"Custom error: {error}")
-    
+
     def error_cb_runtime_error(error):
         """Error callback that raises RuntimeError"""
         raise RuntimeError(f"Runtime error: {error}")
-    
+
     # Test with KafkaException
     consumer1 = TestConsumer({
         'group.id': 'test-kafka-exception',
@@ -430,11 +430,11 @@ def error_cb_runtime_error(error):
         'error_cb': error_cb_kafka_exception
     })
     consumer1.subscribe(['test-topic'])
-    
+
     with pytest.raises(KafkaException):
         consumer1.consume(timeout=0.1)
     consumer1.close()
-    
+
     # Test with ValueError
     consumer2 = TestConsumer({
         'group.id': 'test-value-error',
@@ -444,12 +444,12 @@ def error_cb_runtime_error(error):
         'error_cb': error_cb_value_error
     })
     consumer2.subscribe(['test-topic'])
-    
+
     with pytest.raises(ValueError) as exc_info:
         consumer2.consume(timeout=0.1)
     assert "Custom error:" in str(exc_info.value)
     consumer2.close()
-    
+
     # Test with RuntimeError
     consumer3 = TestConsumer({
         'group.id': 'test-runtime-error',
@@ -459,14 +459,8 @@ def error_cb_runtime_error(error):
         'error_cb': error_cb_runtime_error
     })
     consumer3.subscribe(['test-topic'])
-    
+
     with pytest.raises(RuntimeError) as exc_info:
         consumer3.consume(timeout=0.1)
     assert "Runtime error:" in str(exc_info.value)
     consumer3.close()
-
-
-
-
-
-