do not use protocol version and set_core_connections_per_host in setup/teardown

aboudreault · aboudreault · commit d12f6c0a3b71 · 2016-04-26T09:14:19.000-04:00
diff --git a/benchmarks/base.py b/benchmarks/base.py
@@ -63,8 +63,7 @@
 def setup(hosts):
     log.info("Using 'cassandra' package from %s", cassandra.__path__)
 
-    cluster = Cluster(hosts, protocol_version=1)
-    cluster.set_core_connections_per_host(HostDistance.LOCAL, 1)
+    cluster = Cluster(hosts)
     try:
         session = cluster.connect()
 
@@ -91,8 +90,7 @@ def setup(hosts):
 
 
 def teardown(hosts):
-    cluster = Cluster(hosts, protocol_version=1)
-    cluster.set_core_connections_per_host(HostDistance.LOCAL, 1)
+    cluster = Cluster(hosts)
     session = cluster.connect()
     session.execute("DROP KEYSPACE " + KEYSPACE)
     cluster.shutdown()
