def get_cassandra_connection(cql_version=None): auth_provider = PlainTextAuthProvider(username=TEST_USER, password=TEST_PWD) conn = Cluster((TEST_HOST, ), TEST_PORT, auth_provider=auth_provider, cql_version=cql_version) # until the cql lib does this for us conn.cql_version = cql_version return conn