Skip to content

Commit 3938563

Browse files
authored
Merge pull request #99 from petraselmer/1-0-Add-params-format-queries
Added params and fixed formatting of queries
2 parents e5a4f5f + a71a534 commit 3938563

File tree

1 file changed

+31
-14
lines changed

1 file changed

+31
-14
lines changed

examples/test_examples.py

Lines changed: 31 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,12 @@ def test_minimal_working_example(self):
5151
driver = GraphDatabase.driver("bolt://localhost:7687", auth=basic_auth("neo4j", "neo4j"))
5252
session = driver.session()
5353

54-
session.run("CREATE (a:Person {name:'Arthur', title:'King'})")
54+
session.run("CREATE (a:Person {name: {name}, title: {title}})",
55+
{"name": "Arthur", "title": "King"})
5556

56-
result = session.run("MATCH (a:Person) WHERE a.name = 'Arthur' RETURN a.name AS name, a.title AS title")
57+
result = session.run("MATCH (a:Person) WHERE a.name = {name} "
58+
"RETURN a.name AS name, a.title AS title",
59+
{"name": "Arthur"})
5760
for record in result:
5861
print("%s %s" % (record["title"], record["name"]))
5962

@@ -106,7 +109,8 @@ def test_statement(self):
106109
driver = GraphDatabase.driver("bolt://localhost:7687", auth=auth_token)
107110
session = driver.session()
108111
# tag::statement[]
109-
result = session.run("CREATE (person:Person {name: {name}})", {"name": "Arthur"})
112+
result = session.run("CREATE (person:Person {name: {name}})",
113+
{"name": "Arthur"})
110114
# end::statement[]
111115
result.consume()
112116
session.close()
@@ -126,7 +130,8 @@ def test_result_traversal(self):
126130
# tag::result-traversal[]
127131
search_term = "Sword"
128132
result = session.run("MATCH (weapon:Weapon) WHERE weapon.name CONTAINS {term} "
129-
"RETURN weapon.name", {"term": search_term})
133+
"RETURN weapon.name",
134+
{"term": search_term})
130135
print("List of weapons called %r:" % search_term)
131136
for record in result:
132137
print(record["weapon.name"])
@@ -139,7 +144,8 @@ def test_access_record(self):
139144
# tag::access-record[]
140145
search_term = "Arthur"
141146
result = session.run("MATCH (weapon:Weapon) WHERE weapon.owner CONTAINS {term} "
142-
"RETURN weapon.name, weapon.material, weapon.size", {"term": search_term})
147+
"RETURN weapon.name, weapon.material, weapon.size",
148+
{"term": search_term})
143149
print("List of weapons owned by %r:" % search_term)
144150
for record in result:
145151
print(", ".join("%s: %s" % (key, record[key]) for key in record.keys()))
@@ -151,7 +157,8 @@ def test_result_retention(self):
151157
# tag::retain-result[]
152158
session = driver.session()
153159
result = session.run("MATCH (knight:Person:Knight) WHERE knight.castle = {castle} "
154-
"RETURN knight.name AS name", {"castle": "Camelot"})
160+
"RETURN knight.name AS name",
161+
{"castle": "Camelot"})
155162
retained_result = list(result)
156163
session.close()
157164
for record in retained_result:
@@ -164,11 +171,13 @@ def test_nested_statements(self):
164171
session = driver.session()
165172
# tag::nested-statements[]
166173
result = session.run("MATCH (knight:Person:Knight) WHERE knight.castle = {castle} "
167-
"RETURN id(knight) AS knight_id", {"castle": "Camelot"})
174+
"RETURN id(knight) AS knight_id",
175+
{"castle": "Camelot"})
168176
for record in result:
169177
session.run("MATCH (knight) WHERE id(knight) = {id} "
170178
"MATCH (king:Person) WHERE king.name = {king} "
171-
"CREATE (knight)-[:DEFENDS]->(king)", {"id": record["knight_id"], "king": "Arthur"})
179+
"CREATE (knight)-[:DEFENDS]->(king)",
180+
{"id": record["knight_id"], "king": "Arthur"})
172181
# end::nested-statements[]
173182
session.close()
174183

@@ -177,10 +186,13 @@ def test_transaction_commit(self):
177186
session = driver.session()
178187
# tag::transaction-commit[]
179188
with session.begin_transaction() as tx:
180-
tx.run("CREATE (:Person {name: 'Guinevere'})")
189+
tx.run("CREATE (:Person {name: {name}})",
190+
{"name": "Guinevere"})
181191
tx.success = True
182192
# end::transaction-commit[]
183-
result = session.run("MATCH (p:Person {name: 'Guinevere'}) RETURN count(p)")
193+
result = session.run("MATCH (p:Person {name: {name}}) "
194+
"RETURN count(p)",
195+
{"name": "Guinevere"})
184196
record = next(iter(result))
185197
assert record["count(p)"] == 1
186198
session.close()
@@ -190,10 +202,13 @@ def test_transaction_rollback(self):
190202
session = driver.session()
191203
# tag::transaction-rollback[]
192204
with session.begin_transaction() as tx:
193-
tx.run("CREATE (:Person {name: 'Merlin'})")
205+
tx.run("CREATE (:Person {name: {name}})",
206+
{"name": "Merlin"})
194207
tx.success = False
195208
# end::transaction-rollback[]
196-
result = session.run("MATCH (p:Person {name: 'Merlin'}) RETURN count(p)")
209+
result = session.run("MATCH (p:Person {name: {name}}) "
210+
"RETURN count(p)",
211+
{"name": "Merlin"})
197212
record = next(iter(result))
198213
assert record["count(p)"] == 0
199214
session.close()
@@ -203,7 +218,8 @@ def test_result_summary_query_profile(self):
203218
session = driver.session()
204219
# tag::result-summary-query-profile[]
205220
result = session.run("PROFILE MATCH (p:Person {name: {name}}) "
206-
"RETURN id(p)", {"name": "Arthur"})
221+
"RETURN id(p)",
222+
{"name": "Arthur"})
207223
summary = result.consume()
208224
print(summary.statement_type)
209225
print(summary.profile)
@@ -214,7 +230,8 @@ def test_result_summary_notifications(self):
214230
driver = GraphDatabase.driver("bolt://localhost:7687", auth=auth_token)
215231
session = driver.session()
216232
# tag::result-summary-notifications[]
217-
result = session.run("EXPLAIN MATCH (king), (queen) RETURN king, queen")
233+
result = session.run("EXPLAIN MATCH (king), (queen) "
234+
"RETURN king, queen")
218235
summary = result.consume()
219236
for notification in summary.notifications:
220237
print(notification)

0 commit comments

Comments
 (0)