fixes in test

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/telemetry/telemetry_client.py

@@ -399,7 +399,6 @@ def initialize_telemetry_client(
         host_url,
     ):
         """Initialize a telemetry client for a specific connection if telemetry is enabled"""
-
         try:
             with TelemetryClientFactory._lock:
                 TelemetryClientFactory._initialize()

tests/unit/test_telemetry.py

@@ -235,7 +235,7 @@ def test_export_event(self, telemetry_client_setup):
         client._flush.assert_not_called()
         assert len(client._events_batch) == 5
 
-        for i in range(5):
+        for i in range(5, 10):
             client._export_event(f"event-{i}")
 
         client._flush.assert_called_once()