@@ -63,12 +63,10 @@ def cassandra_session(cassandra_cluster):
6363 session .execute ("DROP KEYSPACE testkeyspace;" )
6464 except ConfigurationException :
6565 pass
66- session .execute (
67- """
66+ session .execute ("""
6867 CREATE KEYSPACE testkeyspace
6968 WITH REPLICATION = { 'class' : 'SimpleStrategy' ,'replication_factor' : 1 }
70- """
71- )
69+ """ )
7270 session .execute ("USE testkeyspace;" )
7371 session .execute ("CREATE TABLE testkeyspace.users ( id UUID PRIMARY KEY, name text);" )
7472 yield session
@@ -98,12 +96,10 @@ def test_cassandra_connect(instrument, elasticapm_client, cassandra_cluster):
9896def test_cassandra_connect_keyspace (instrument , elasticapm_client , cassandra_cluster ):
9997 session = cassandra_cluster .connect ()
10098 try :
101- session .execute (
102- """
99+ session .execute ("""
103100 CREATE KEYSPACE testkeyspace
104101 WITH REPLICATION = { 'class' : 'SimpleStrategy' ,'replication_factor' : 1 }
105- """
106- )
102+ """ )
107103 elasticapm_client .begin_transaction ("transaction.test" )
108104 sess = cassandra_cluster .connect ("testkeyspace" )
109105 elasticapm_client .end_transaction ("test" )
@@ -185,19 +181,14 @@ def test_signature_create_keyspace():
185181
186182
187183def test_signature_create_columnfamily ():
188- assert (
189- extract_signature (
190- """CREATE COLUMNFAMILY users (
184+ assert extract_signature ("""CREATE COLUMNFAMILY users (
191185 userid text PRIMARY KEY,
192186 first_name text,
193187 last_name text,
194188 emails set<text>,
195189 top_scores list<int>,
196190 todo map<timestamp, text>
197- );"""
198- )
199- == "CREATE COLUMNFAMILY"
200- )
191+ );""" ) == "CREATE COLUMNFAMILY"
201192
202193
203194def test_select_from_collection ():
0 commit comments