diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py index 3ea397f0..8933eefd 100644 --- a/tests/integration/__init__.py +++ b/tests/integration/__init__.py @@ -118,11 +118,11 @@ else: log.info('Using Cassandra version: %s', CASSANDRA_VERSION) CCM_KWARGS['version'] = CASSANDRA_VERSION -if CASSANDRA_VERSION > '3.0': +if CASSANDRA_VERSION >= '3.0': default_protocol_version = 4 -elif CASSANDRA_VERSION > '2.1': +elif CASSANDRA_VERSION >= '2.1': default_protocol_version = 3 -elif CASSANDRA_VERSION > '2.0': +elif CASSANDRA_VERSION >= '2.0': default_protocol_version = 2 else: default_protocol_version = 1