Remove unused imports from tests

mykaul · mykaul · commit e5c83c38f629 · 2026-06-03T11:06:40.000+03:00
diff --git a/tests/integration/cqlengine/__init__.py b/tests/integration/cqlengine/__init__.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import os
-import unittest
 from cassandra import ConsistencyLevel
 
 from cassandra.cqlengine import connection
diff --git a/tests/integration/cqlengine/base.py b/tests/integration/cqlengine/base.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 import unittest
 
-import sys
 
 from cassandra.cqlengine.connection import get_session
 from cassandra.cqlengine.models import Model
diff --git a/tests/integration/cqlengine/connections/test_connection.py b/tests/integration/cqlengine/connections/test_connection.py
@@ -23,7 +23,7 @@
 from cassandra.policies import RoundRobinPolicy
 from cassandra.query import dict_factory
 
-from tests.integration import CASSANDRA_IP, PROTOCOL_VERSION, execute_with_long_wait_retry, local, TestCluster
+from tests.integration import CASSANDRA_IP, execute_with_long_wait_retry, local, TestCluster
 from tests.integration.cqlengine.base import BaseCassEngTestCase
 from tests.integration.cqlengine import DEFAULT_KEYSPACE, setup_connection
 
diff --git a/tests/integration/cqlengine/management/test_management.py b/tests/integration/cqlengine/management/test_management.py
@@ -14,7 +14,6 @@
 import unittest
 
 from unittest import mock
-import logging
 from packaging.version import Version
 from cassandra.cqlengine.connection import get_session, get_cluster
 from cassandra.cqlengine import CQLEngineException
diff --git a/tests/integration/cqlengine/model/test_model.py b/tests/integration/cqlengine/model/test_model.py
@@ -20,7 +20,6 @@
 from cassandra.cqlengine import models
 from cassandra.cqlengine.models import Model, ModelDefinitionException
 from uuid import uuid1
-from tests.integration import pypy
 from tests.integration.cqlengine.base import TestQueryUpdateModel
 import pytest
 
diff --git a/tests/integration/cqlengine/model/test_model_io.py b/tests/integration/cqlengine/model/test_model_io.py
@@ -19,7 +19,6 @@
 from decimal import Decimal
 from operator import itemgetter
 
-import cassandra
 from cassandra.cqlengine import columns
 from cassandra.cqlengine import CQLEngineException
 from cassandra.cqlengine.management import sync_table
diff --git a/tests/integration/cqlengine/query/test_named.py b/tests/integration/cqlengine/query/test_named.py
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import unittest
 
 from cassandra import ConsistencyLevel
 from cassandra.cqlengine import operators
@@ -22,7 +21,7 @@
 from cassandra.concurrent import execute_concurrent_with_args
 from cassandra.cqlengine import models
 
-from tests.integration.cqlengine import setup_connection, execute_count
+from tests.integration.cqlengine import execute_count
 from tests.integration.cqlengine.base import BaseCassEngTestCase
 from tests.integration.cqlengine.query.test_queryset import BaseQuerySetUsage
 
diff --git a/tests/integration/cqlengine/statements/test_select_statement.py b/tests/integration/cqlengine/statements/test_select_statement.py
@@ -14,7 +14,7 @@
 import unittest
 
 from cassandra.cqlengine.columns import Column
-from cassandra.cqlengine.statements import SelectStatement, WhereClause
+from cassandra.cqlengine.statements import SelectStatement
 from cassandra.cqlengine.operators import *
 
 class SelectStatementTests(unittest.TestCase):
diff --git a/tests/integration/cqlengine/statements/test_update_statement.py b/tests/integration/cqlengine/statements/test_update_statement.py
@@ -15,9 +15,7 @@
 
 from cassandra.cqlengine.columns import Column, Set, List, Text
 from cassandra.cqlengine.operators import *
-from cassandra.cqlengine.statements import (UpdateStatement, WhereClause,
-                                  AssignmentClause, SetUpdateClause,
-                                  ListUpdateClause)
+from cassandra.cqlengine.statements import (UpdateStatement)
 
 
 class UpdateStatementTests(unittest.TestCase):
diff --git a/tests/integration/cqlengine/test_connections.py b/tests/integration/cqlengine/test_connections.py
@@ -17,12 +17,12 @@
 from cassandra.cqlengine import columns, CQLEngineException
 from cassandra.cqlengine import connection as conn
 from cassandra.cqlengine.management import drop_keyspace, sync_table, drop_table, create_keyspace_simple
-from cassandra.cqlengine.models import Model, QuerySetDescriptor
+from cassandra.cqlengine.models import Model
 from cassandra.cqlengine.query import ContextQuery, BatchQuery, ModelQuerySet
 from tests.integration.cqlengine import setup_connection, DEFAULT_KEYSPACE
 from tests.integration.cqlengine.base import BaseCassEngTestCase
 from tests.integration.cqlengine.query import test_queryset
-from tests.integration import local, CASSANDRA_IP, TestCluster
+from tests.integration import CASSANDRA_IP, TestCluster
 import pytest
 
 
diff --git a/tests/integration/cqlengine/test_ifexists.py b/tests/integration/cqlengine/test_ifexists.py
@@ -18,7 +18,7 @@
 from cassandra.cqlengine import columns
 from cassandra.cqlengine.management import sync_table, drop_table
 from cassandra.cqlengine.models import Model
-from cassandra.cqlengine.query import BatchQuery, BatchType, LWTException, IfExistsWithCounterColumn
+from cassandra.cqlengine.query import BatchQuery, LWTException, IfExistsWithCounterColumn
 
 from tests.integration.cqlengine.base import BaseCassEngTestCase
 from tests.integration import PROTOCOL_VERSION
diff --git a/tests/integration/cqlengine/test_ttl.py b/tests/integration/cqlengine/test_ttl.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 
-import unittest
 
 from packaging.version import Version
 
@@ -25,7 +24,7 @@
 from cassandra.cqlengine import columns
 from unittest import mock
 from cassandra.cqlengine.connection import get_session
-from tests.integration import CASSANDRA_VERSION, greaterthancass20
+from tests.integration import CASSANDRA_VERSION
 
 
 class TestTTLModel(Model):
diff --git a/tests/integration/long/test_large_data.py b/tests/integration/long/test_large_data.py
@@ -21,7 +21,7 @@
 from cassandra.cluster import ExecutionProfile, EXEC_PROFILE_DEFAULT
 from cassandra.query import dict_factory
 from cassandra.query import SimpleStatement
-from tests.integration import use_singledc, PROTOCOL_VERSION, TestCluster
+from tests.integration import use_singledc, TestCluster
 from tests.integration.long.utils import create_schema
 
 import unittest
diff --git a/tests/integration/simulacron/__init__.py b/tests/integration/simulacron/__init__.py
@@ -18,7 +18,6 @@
     clear_queries,
     start_and_prime_singledc,
     stop_simulacron,
-    start_and_prime_cluster_defaults,
 )
 
 from cassandra.cluster import Cluster
diff --git a/tests/integration/simulacron/test_cluster.py b/tests/integration/simulacron/test_cluster.py
@@ -11,15 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import unittest
 
-import logging
-from packaging.version import Version
 
 import cassandra
-from tests.integration.simulacron import SimulacronCluster, SimulacronBase
+from tests.integration.simulacron import SimulacronCluster
 from tests.integration import (requiressimulacron, PROTOCOL_VERSION, MockLoggingHandler)
-from tests.integration.simulacron.utils import prime_query, start_and_prime_singledc
+from tests.integration.simulacron.utils import prime_query
 
 from cassandra import (WriteTimeout, WriteType,
                        ConsistencyLevel, UnresolvableContactPoints)
diff --git a/tests/integration/simulacron/test_connection.py b/tests/integration/simulacron/test_connection.py
@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import unittest
 
 import logging
 import time
diff --git a/tests/integration/simulacron/test_empty_column.py b/tests/integration/simulacron/test_empty_column.py
@@ -11,11 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import unittest
 
 from collections import namedtuple, OrderedDict
 
-from cassandra import ProtocolVersion
 from cassandra.cluster import Cluster, EXEC_PROFILE_DEFAULT
 from cassandra.query import (named_tuple_factory, tuple_factory,
                              dict_factory, ordered_dict_factory)
diff --git a/tests/integration/simulacron/test_endpoint.py b/tests/integration/simulacron/test_endpoint.py
@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import unittest
 
 from functools import total_ordering
 
diff --git a/tests/integration/simulacron/utils.py b/tests/integration/simulacron/utils.py
@@ -14,7 +14,6 @@
 
 import json
 import subprocess
-import time
 from urllib.request import build_opener, Request, HTTPHandler
 
 from cassandra.metadata import SchemaParserV4, SchemaParserDSE68
diff --git a/tests/integration/standard/conftest.py b/tests/integration/standard/conftest.py
@@ -1,4 +1,3 @@
-import pytest
 import logging
 
 # Cluster topology groups for test ordering.
diff --git a/tests/integration/standard/test_cluster.py b/tests/integration/standard/test_cluster.py
@@ -27,10 +27,10 @@
 import pytest
 
 import cassandra
-from cassandra.cluster import NoHostAvailable, ExecutionProfile, EXEC_PROFILE_DEFAULT, ControlConnection, Cluster
+from cassandra.cluster import NoHostAvailable, ExecutionProfile, EXEC_PROFILE_DEFAULT, Cluster
 from cassandra.concurrent import execute_concurrent
 from cassandra.policies import (RoundRobinPolicy, ExponentialReconnectionPolicy,
-                                RetryPolicy, SimpleConvictionPolicy, HostDistance,
+                                SimpleConvictionPolicy, HostDistance,
                                 AddressTranslator, TokenAwarePolicy, HostFilterPolicy)
 from cassandra import ConsistencyLevel
 
@@ -43,7 +43,7 @@
 from tests.integration import use_cluster, get_server_versions, CASSANDRA_VERSION, \
     execute_until_pass, execute_with_long_wait_retry, get_node, MockLoggingHandler, get_unsupported_lower_protocol, \
     get_unsupported_upper_protocol, local, CASSANDRA_IP, greaterthanorequalcass30, \
-    lessthanorequalcass40, TestCluster, PROTOCOL_VERSION, xfail_scylla, incorrect_test
+    lessthanorequalcass40, TestCluster, PROTOCOL_VERSION, incorrect_test
 from tests.integration.util import assert_quiescent_pool_state
 from tests.util import assertListEqual
 import sys
diff --git a/tests/integration/standard/test_concurrent.py b/tests/integration/standard/test_concurrent.py
@@ -19,7 +19,6 @@
     ReadFailure, WriteFailure
 from cassandra.cluster import ExecutionProfile, EXEC_PROFILE_DEFAULT
 from cassandra.concurrent import execute_concurrent, execute_concurrent_with_args, ExecutionResult
-from cassandra.policies import HostDistance
 from cassandra.query import dict_factory, tuple_factory, SimpleStatement
 
 from tests.integration import use_singledc, PROTOCOL_VERSION, TestCluster
diff --git a/tests/integration/standard/test_concurrent_schema_change_and_node_kill.py b/tests/integration/standard/test_concurrent_schema_change_and_node_kill.py
@@ -1,4 +1,3 @@
-import os
 import logging
 import unittest
 
diff --git a/tests/integration/standard/test_custom_cluster.py b/tests/integration/standard/test_custom_cluster.py
@@ -14,7 +14,7 @@
 
 from cassandra.cluster import NoHostAvailable
 from tests.integration import use_singledc, get_cluster, remove_cluster, local, TestCluster
-from tests.util import wait_until, wait_until_not_raised
+from tests.util import wait_until
 
 import unittest
 import pytest
diff --git a/tests/integration/standard/test_custom_payload.py b/tests/integration/standard/test_custom_payload.py
@@ -17,7 +17,7 @@
 
 from cassandra.query import (SimpleStatement, BatchStatement, BatchType)
 
-from tests.integration import (use_singledc, PROTOCOL_VERSION, local, TestCluster,
+from tests.integration import (use_singledc, local, TestCluster,
                               requires_custom_payload)
 import pytest
 
diff --git a/tests/integration/standard/test_custom_protocol_handler.py b/tests/integration/standard/test_custom_protocol_handler.py
@@ -16,19 +16,16 @@
 
 from cassandra.protocol import ProtocolHandler, ResultMessage, QueryMessage, UUIDType, read_int
 from cassandra.query import tuple_factory, SimpleStatement
-from cassandra.cluster import (ResponseFuture, ExecutionProfile, EXEC_PROFILE_DEFAULT,
-    ContinuousPagingOptions, NoHostAvailable)
+from cassandra.cluster import (ResponseFuture, ExecutionProfile, EXEC_PROFILE_DEFAULT)
 from cassandra import ProtocolVersion, ConsistencyLevel
 
 from tests.integration import use_single_node, drop_keyspace_shutdown_cluster, \
-    greaterthanorequalcass30, execute_with_long_wait_retry, greaterthanorequalcass3_10, \
-    TestCluster, greaterthanorequalcass40
+    greaterthanorequalcass30, execute_with_long_wait_retry, TestCluster, greaterthanorequalcass40
 from tests.integration.datatype_utils import update_datatypes, PRIMITIVE_DATATYPES
 from tests.integration.standard.utils import create_table_with_all_types, get_all_primitive_params
 
 import uuid
 from unittest import mock
-import pytest
 
 
 def setup_module():
diff --git a/tests/integration/standard/test_metadata.py b/tests/integration/standard/test_metadata.py
@@ -14,38 +14,31 @@
 
 import unittest
 
-from collections import defaultdict
-import difflib
 import logging
 import sys
 import time
 import os
-from typing import Optional
 
 from packaging.version import Version
 from unittest.mock import Mock, patch
 import pytest
 
 from cassandra import AlreadyExists, SignatureDescriptor, UserFunctionDescriptor, UserAggregateDescriptor
-from cassandra.connection import Connection
 
 from cassandra.encoder import Encoder
 from cassandra.metadata import (IndexMetadata, Token, murmur3, Function, Aggregate, protect_name, protect_names,
                                 RegisteredTableExtension, _RegisteredExtensionType, get_schema_parser,
                                 group_keys_by_replica, NO_VALID_REPLICA)
 from cassandra.protocol import QueryMessage, ProtocolHandler
-from cassandra.util import SortedSet
 
 from tests.integration import (get_cluster, use_singledc, PROTOCOL_VERSION, execute_until_pass,
                                BasicSegregatedKeyspaceUnitTestCase, BasicSharedKeyspaceUnitTestCase,
                                BasicExistingKeyspaceUnitTestCase, drop_keyspace_shutdown_cluster, CASSANDRA_VERSION,
                                greaterthanorequalcass30, lessthancass30, local,
                                get_supported_protocol_versions, greaterthancass20,
-                               greaterthancass21, greaterthanorequalcass40,
-                               lessthancass40,
+                               greaterthancass21, lessthancass40,
                                TestCluster, requires_java_udf, requires_composite_type,
-                               requires_collection_indexes, SCYLLA_VERSION, xfail_scylla, xfail_scylla_version_lt,
-                               requirescompactstorage, get_tablets_disabled_ddl_suffix, execute_with_long_wait_retry)
+                               requires_collection_indexes, SCYLLA_VERSION, xfail_scylla, requirescompactstorage, get_tablets_disabled_ddl_suffix, execute_with_long_wait_retry)
 
 from tests.util import wait_until, assertRegex, assertDictEqual, assertListEqual, assert_startswith_diff
 
diff --git a/tests/integration/standard/test_policies.py b/tests/integration/standard/test_policies.py
@@ -15,9 +15,7 @@
 import unittest
 
 from cassandra.cluster import ExecutionProfile, EXEC_PROFILE_DEFAULT
-from cassandra.policies import HostFilterPolicy, RoundRobinPolicy,  SimpleConvictionPolicy, \
-    WhiteListRoundRobinPolicy, ExponentialBackoffRetryPolicy, ColDesc
-from cassandra.pool import Host
+from cassandra.policies import HostFilterPolicy, RoundRobinPolicy,  WhiteListRoundRobinPolicy, ExponentialBackoffRetryPolicy
 from cassandra.connection import DefaultEndPoint
 
 from tests.integration import local, use_singledc, TestCluster
diff --git a/tests/integration/standard/test_query.py b/tests/integration/standard/test_query.py
@@ -23,7 +23,7 @@
 from cassandra.query import (PreparedStatement, BoundStatement, SimpleStatement,
                              BatchStatement, BatchType, dict_factory, TraceUnavailable)
 from cassandra.cluster import NoHostAvailable, ExecutionProfile, EXEC_PROFILE_DEFAULT, Cluster
-from cassandra.policies import HostDistance, RoundRobinPolicy, WhiteListRoundRobinPolicy
+from cassandra.policies import RoundRobinPolicy, WhiteListRoundRobinPolicy
 from tests.integration import use_singledc, PROTOCOL_VERSION, BasicSharedKeyspaceUnitTestCase, \
     greaterthanprotocolv3, MockLoggingHandler, get_supported_protocol_versions, local, get_cluster, setup_keyspace, \
     USE_CASS_EXTERNAL, greaterthanorequalcass40, TestCluster, xfail_scylla, xfail_scylla_version_lt, \
diff --git a/tests/integration/standard/test_query_paging.py b/tests/integration/standard/test_query_paging.py
@@ -25,7 +25,6 @@
 from cassandra import ConsistencyLevel
 from cassandra.cluster import EXEC_PROFILE_DEFAULT, ExecutionProfile
 from cassandra.concurrent import execute_concurrent, execute_concurrent_with_args
-from cassandra.policies import HostDistance
 from cassandra.query import SimpleStatement
 
 from tests.util import assertSequenceEqual
diff --git a/tests/integration/standard/test_shard_aware.py b/tests/integration/standard/test_shard_aware.py
@@ -23,7 +23,7 @@
 
 from cassandra.cluster import Cluster
 from cassandra.policies import TokenAwarePolicy, RoundRobinPolicy, ConstantReconnectionPolicy
-from cassandra import OperationTimedOut, ConsistencyLevel
+from cassandra import OperationTimedOut
 
 from tests.integration import use_cluster, get_node, PROTOCOL_VERSION
 from tests.util import wait_until_not_raised
diff --git a/tests/integration/standard/test_single_interface.py b/tests/integration/standard/test_single_interface.py
@@ -13,14 +13,12 @@
 # limitations under the License.
 
 import unittest
-import pytest
 
 from cassandra import ConsistencyLevel
 from cassandra.query import SimpleStatement
 
 from packaging.version import Version
-from tests.integration import use_singledc, PROTOCOL_VERSION, \
-    remove_cluster, greaterthanorequalcass40, \
+from tests.integration import use_singledc, remove_cluster, greaterthanorequalcass40, \
     CASSANDRA_VERSION, TestCluster, DEFAULT_SINGLE_INTERFACE_PORT
 
 
diff --git a/tests/integration/standard/test_types.py b/tests/integration/standard/test_types.py
@@ -21,15 +21,13 @@
 import socket
 import uuid
 
-from datetime import datetime, date, time, timedelta
+from datetime import datetime, timedelta
 from decimal import Decimal
 from functools import partial
 
-from packaging.version import Version
 
 import cassandra
 from cassandra import InvalidRequest
-from cassandra import util
 from cassandra.cluster import ExecutionProfile, EXEC_PROFILE_DEFAULT
 from cassandra.concurrent import execute_concurrent_with_args
 from cassandra.cqltypes import Int32Type, EMPTY
diff --git a/tests/integration/upgrade/__init__.py b/tests/integration/upgrade/__init__.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-from tests.integration import CCM_KWARGS, use_cluster, remove_cluster, MockLoggingHandler
+from tests.integration import use_cluster, remove_cluster, MockLoggingHandler
 from tests.integration import setup_keyspace
 
 from cassandra.cluster import Cluster
diff --git a/tests/integration/upgrade/test_upgrade.py b/tests/integration/upgrade/test_upgrade.py
diff --git a/tests/stress_tests/test_load.py b/tests/stress_tests/test_load.py
diff --git a/tests/unit/advanced/test_graph.py b/tests/unit/advanced/test_graph.py
diff --git a/tests/unit/advanced/test_insights.py b/tests/unit/advanced/test_insights.py
diff --git a/tests/unit/advanced/test_metadata.py b/tests/unit/advanced/test_metadata.py
diff --git a/tests/unit/io/test_asyncorereactor.py b/tests/unit/io/test_asyncorereactor.py
diff --git a/tests/unit/io/test_twistedreactor.py b/tests/unit/io/test_twistedreactor.py
diff --git a/tests/unit/test_endpoints.py b/tests/unit/test_endpoints.py
diff --git a/tests/unit/test_host_connection_pool.py b/tests/unit/test_host_connection_pool.py
diff --git a/tests/unit/test_marshalling.py b/tests/unit/test_marshalling.py
diff --git a/tests/unit/test_protocol.py b/tests/unit/test_protocol.py
