def test07_test_edge_filters(self):
node0 = Node(node_id=0, label="L")
node1 = Node(node_id=1, label="L")
node2 = Node(node_id=2, label="L")
edge01 = Edge(src_node=node0, dest_node=node1, relation="R", properties={'x':1})
edge12 = Edge(src_node=node1, dest_node=node2, relation="R")
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_node(node2)
redis_graph.add_edge(edge01)
redis_graph.add_edge(edge12)
redis_graph.flush()
query = "MATCH (n:L) WHERE (n)-[:R {x:1}]->() RETURN n"
result_set = redis_graph.query(query)
expected_results = [[node0]]
query_info = QueryInfo(query = query, description="Tests pattern filter edge conditions", expected_result = expected_results)
self._assert_resultset_and_expected_mutually_included(result_set, query_info)
def test02_test_path_filter_or_property_filter(self):
node0 = Node(node_id=0, label="L")
node1 = Node(node_id=1, label="L", properties={'x': 1})
edge01 = Edge(src_node=node0, dest_node=node1, relation="R")
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_edge(edge01)
redis_graph.flush()
query = "MATCH (n:L) WHERE (n)-[:R]->(:L) OR n.x=1 RETURN n"
result_set = redis_graph.query(query)
expected_results = [[node0], [node1]]
query_info = QueryInfo(
query=query,
description="Tests OR condition with simple filter and path filter",
expected_result=expected_results)
self._assert_resultset_and_expected_mutually_included(
result_set, query_info)
def test_parameterized_skip_limit(self):
params = {'skip': 1, 'limit': 1}
query = "UNWIND [1,2,3] AS X RETURN X SKIP $skip LIMIT $limit"
expected_results = [[2]]
query_info = QueryInfo(query=query,
description="Tests skip limit as params",
expected_result=expected_results)
self._assert_resultset_equals_expected(
redis_graph.query(query, params), query_info)
# Set one parameter to non-integer value
params = {'skip': '1', 'limit': 1}
try:
redis_graph.query(query, params)
assert (False)
except redis.exceptions.ResponseError as e:
pass
def test_node_retrival(self):
p0 = Node(node_id=0, label="Person", properties={'name': 'a'})
p1 = Node(node_id=1, label="Person", properties={'name': 'b'})
p2 = Node(node_id=2, label="NoPerson", properties={'name': 'a'})
redis_graph.add_node(p0)
redis_graph.add_node(p1)
redis_graph.add_node(p2)
redis_graph.flush()
params = {'name': 'a'}
query = "MATCH (n :Person {name:$name}) RETURN n"
expected_results = [[p0]]
query_info = QueryInfo(query=query,
description="Tests expression on param",
expected_result=expected_results)
self._assert_resultset_equals_expected(
redis_graph.query(query, params), query_info)
def test06_test_level_2_nesting_logical_operators_over_path_filters(self):
node0 = Node(node_id=0, label="L")
node1 = Node(node_id=1, label="L", properties={'x':1})
node2 = Node(node_id=2, label="L2")
edge01 = Edge(src_node=node0, dest_node=node1, relation="R")
edge12 = Edge(src_node=node1, dest_node=node2, relation="R2")
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_node(node2)
redis_graph.add_edge(edge01)
redis_graph.add_edge(edge12)
redis_graph.flush()
query = "MATCH (n:L) WHERE (n)-[:R]->(:L) OR (n.x=1 AND ((n)-[:R2]->(:L2) OR (n)-[:R]->(:L))) RETURN n"
result_set = redis_graph.query(query)
expected_results = [[node0],[node1]]
query_info = QueryInfo(query = query, description="Tests AND condition with simple filter and nested OR", expected_result = expected_results)
self._assert_resultset_and_expected_mutually_included(result_set, query_info)
def test_bi_directional_path(self):
node0 = Node(node_id=0, label="L1")
node1 = Node(node_id=1, label="L1")
node2 = Node(node_id=2, label="L1")
edge01 = Edge(node0, "R1", node1, edge_id=0, properties={'value': 1})
edge12 = Edge(node1, "R1", node2, edge_id=1, properties={'value': 2})
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_node(node2)
redis_graph.add_edge(edge01)
redis_graph.add_edge(edge12)
redis_graph.flush()
# Rewrite the edges with IDs instead of node values to match how they are returned.
edge01 = Edge(0, "R1", 1, edge_id=0, properties={'value': 1})
edge12 = Edge(1, "R1", 2, edge_id=1, properties={'value': 2})
# Reverse direction edges which are not part of the graph. Read only values.
edge10 = Edge(1, "R1", 0, edge_id=0, properties={'value': 1})
edge21 = Edge(2, "R1", 1, edge_id=1, properties={'value': 2})
path010 = Path.new_empty_path().add_node(node0).add_edge(
edge01).add_node(node1).add_edge(edge10).add_node(node0)
path0121 = Path.new_empty_path().add_node(node0).add_edge(edge01).add_node(node1).add_edge(edge12) \
.add_node(node2).add_edge(edge21).add_node(node1)
path01210 = Path.new_empty_path().add_node(node0).add_edge(edge01).add_node(node1).add_edge(edge12) \
.add_node(node2).add_edge(edge21).add_node(node1).add_edge(edge10).add_node(node0)
path121 = Path.new_empty_path().add_node(node1).add_edge(
edge12).add_node(node2).add_edge(edge21).add_node(node1)
path1210 = Path.new_empty_path().add_node(node1).add_edge(edge12).add_node(node2).add_edge(edge21) \
.add_node(node1).add_edge(edge10).add_node(node0)
expected_results = [[path010], [path0121], [path01210], [path121],
[path1210]]
query = "MATCH p=(:L1)-[:R1*]->(:L1)<-[:R1*]-() RETURN p"
query_info = QueryInfo(
query=query,
description="Tests bi directional variable length paths",
expected_result=expected_results)
self._assert_resultset_and_expected_mutually_included(
redis_graph.query(query), query_info)
def test_zero_length_path(self):
node0 = Node(node_id=0, label="L1")
node1 = Node(node_id=1, label="L2")
edge01 = Edge(node0, "R1", node1, edge_id=0, properties={'value': 1})
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_edge(edge01)
redis_graph.flush()
path01 = Path.new_empty_path().add_node(node0).add_edge(edge01).add_node(node1)
expected_results=[[path01]]
query = "MATCH p=(:L1)-[*0..]->()-[]->(:L2) RETURN p"
query_info = QueryInfo(query = query, description="Tests path with zero length variable length paths", \
expected_result = expected_results)
self._assert_resultset_and_expected_mutually_included(redis_graph.query(query), query_info)
def test00_test_data_valid_after_rename(self):
global graph
node0 = Node(node_id=0, label="L", properties={'name': 'x', 'age': 1})
graph.add_node(node0)
graph.flush()
redis_con.rename(GRAPH_ID, NEW_GRAPH_ID)
graph = Graph(redis_con, NEW_GRAPH_ID)
node1 = Node(node_id=1, label="L", properties={'name': 'x', 'age': 1})
graph.add_node(node1)
graph.flush()
query = "MATCH (n) return n"
expected_results = [[node0], [node1]]
query_info = QueryInfo(
query=query,
description="Tests data is valid after renaming",
expected_result=expected_results)
self._assert_resultset_and_expected_mutually_included(
graph.query(query), query_info)
def test_simple_path(self):
node0 = Node(node_id=0, label="L1")
node1 = Node(node_id=1, label="L1")
node2 = Node(node_id=2, label="L1")
edge01 = Edge(node0, "R1", node1, edge_id=0, properties={'value': 1})
edge12 = Edge(node1, "R1", node2, edge_id=1, properties={'value': 2})
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_node(node2)
redis_graph.add_edge(edge01)
redis_graph.add_edge(edge12)
redis_graph.flush()
path01 = Path.new_empty_path().add_node(node0).add_edge(edge01).add_node(node1)
path12 = Path.new_empty_path().add_node(node1).add_edge(edge12).add_node(node2)
expected_results = [[path01], [path12]]
query = "MATCH p=(:L1)-[:R1]->(:L1) RETURN p"
query_info = QueryInfo(query = query, description="Tests simple paths", expected_result = expected_results)
self._assert_resultset_and_expected_mutually_included(redis_graph.query(query), query_info)
def test_bi_directional_path_functions(self):
node0 = Node(node_id=0, label="L1")
node1 = Node(node_id=1, label="L1")
node2 = Node(node_id=2, label="L1")
edge01 = Edge(node0, "R1", node1, edge_id=0, properties={'value': 1})
edge12 = Edge(node1, "R1", node2, edge_id=1, properties={'value': 2})
redis_graph.add_node(node0)
redis_graph.add_node(node1)
redis_graph.add_node(node2)
redis_graph.add_edge(edge01)
redis_graph.add_edge(edge12)
redis_graph.flush()
# Rewrite the edges with IDs instead of node values to match how they are returned.
edge01 = Edge(0, "R1", 1, edge_id=0, properties={'value': 1})
edge12 = Edge(1, "R1", 2, edge_id=1, properties={'value': 2})
# Reverse direction edges which are not part of the graph. Read only values.
edge10 = Edge(1, "R1", 0, edge_id=0, properties={'value': 1})
edge21 = Edge(2, "R1", 1, edge_id=1, properties={'value': 2})
expected_results = [[[node0, node1, node0], [edge01, edge10], 2],
[[node0, node1, node2, node1],
[edge01, edge12, edge21], 3],
[[node0, node1, node2, node1, node0],
[edge01, edge12, edge21, edge10], 4],
[[node1, node2, node1], [edge12, edge21], 2],
[[node1, node2, node1, node0],
[edge12, edge21, edge10], 3]]
query = "MATCH p=(:L1)-[:R1*]->(:L1)<-[:R1*]-() RETURN nodes(p), relationships(p), length(p)"
query_info = QueryInfo(query = query, description="Tests path functions over bi directional variable length paths", \
expected_result = expected_results)
self._assert_resultset_and_expected_mutually_included(
redis_graph.query(query), query_info)
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
my_friends_query = QueryInfo(
query="""MATCH (ME:person {name:"Roi Lipman"})-[:friend]->(f:person)
RETURN f.name""",
description='My friends?',
max_run_time_ms=0.25,
expected_result=[['"Tal Doron"'], ['"Omri Traub"'], ['"Boaz Arad"'],
['"Ori Laslo"'], ['"Ailon Velger"'], ['"Alon Fital"']])
friends_of_friends_query = QueryInfo(
query=
"""MATCH (ME:person {name:"Roi Lipman"})-[:friend]->(:person)-[:friend]->(fof:person)
RETURN fof.name""",
description='Friends of friends?',
max_run_time_ms=0.25,
expected_result=[['"Valerie Abigail Arad"'], ['"Shelly Laslo Rooz"'],
['"Noam Nativ"'], ['"Jane Chernomorin"'],
['"Mor Yesharim"'], ['"Gal Derriere"'],
['"Lucy Yanfital"']])
friends_of_friends_single_and_over_30_query = QueryInfo(
query=
"""MATCH (ME:person {name:"Roi Lipman"})-[:friend]->(:person)-[:friend]->(fof:person {status:single})
WHERE fof.age > 30
RETURN fof""",
description='Friends of friends who are single and over 30?',
def __init__(self, actors=None, movies=None):
nodesAvailable = (movies is not None and actors is not None)
##################################################################
### number_of_actors_query
##################################################################
self.number_of_actors_query = QueryInfo(
query="""MATCH (n:actor) RETURN count(n) as actors_count""",
description='How many actors are in the graph?',
max_run_time_ms=0.2,
expected_result=[[1317]])
##################################################################
### actors_played_with_nicolas_cage_query
##################################################################
self.actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (n:actor{name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title""",
description='Which actors played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Quinton Aaron', 'Left Behind'],
['Martin Klebba', 'Left Behind'],
['Lea Thompson', 'Left Behind'],
['Nicolas Cage', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Jordin Sparks', 'Left Behind']])
##################################################################
### find_three_actors_played_with_nicolas_cage_query
##################################################################
self.find_three_actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (nicolas:actor {name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title
LIMIT 3""",
description='Get 3 actors who have played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Quinton Aaron', 'Left Behind'],
['Martin Klebba', 'Left Behind'],
['Lea Thompson', 'Left Behind'],
['Nicolas Cage', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Jordin Sparks', 'Left Behind']])
##################################################################
### actors_played_in_movie_straight_outta_compton_query
##################################################################
self.actors_played_in_movie_straight_outta_compton_query = QueryInfo(
query=
"""MATCH (a:actor)-[:act]->(m:movie {title:"Straight Outta Compton"})
RETURN a.name""",
description=
'Which actors played in the movie Straight Outta Compton?',
max_run_time_ms=3.5,
expected_result=[['Aldis Hodge'], ['Corey Hawkins'],
['Neil Brown Jr.'], ['O\'Shea Jackson Jr.']])
##################################################################
### actors_over_50_that_played_in_blockbusters_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [
[actors['Bill Irwin'], movies['Interstellar']],
[actors['Vincent Price'], movies['Vincent']],
[actors['Ellen Burstyn'], movies['Interstellar']],
[actors['Paul Reiser'], movies['Whiplash']],
[actors['Francis X. McCarthy'], movies['Interstellar']],
[actors['John Lithgow'], movies['Interstellar']],
[actors['J.K. Simmons'], movies['Whiplash']],
[actors['Chris Mulkey'], movies['Whiplash']],
[actors['Rachael Harris'], movies['Lucifer']],
[actors['Matthew McConaughey'], movies['Interstellar']],
[actors['D.B. Woodside'], movies['Lucifer']]
]
self.actors_over_50_that_played_in_blockbusters_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m:movie)
WHERE a.age >= 50 AND m.votes > 10000 AND m.rating > 8.2
RETURN *""",
description=
'Which actors who are over 50 played in blockbuster movies?',
max_run_time_ms=4.0,
expected_result=expected_result)
##################################################################
### actors_played_in_bad_drama_or_comedy_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [
['Rita Ora', movies['Fifty Shades of Grey']],
['Dakota Johnson', movies['Fifty Shades of Grey']],
['Marcia Gay Harden', movies['Fifty Shades of Grey']],
['Jamie Dornan', movies['Fifty Shades of Grey']],
['Eloise Mumford', movies['Fifty Shades of Grey']],
['Max Martini', movies['Fifty Shades of Grey']],
['Luke Grimes', movies['Fifty Shades of Grey']],
['Jennifer Ehle', movies['Fifty Shades of Grey']],
['Victor Rasuk', movies['Fifty Shades of Grey']],
['Nancy Lenehan', movies['Sex Tape']],
['Rob Lowe', movies['Sex Tape']],
['Cameron Diaz', movies['Sex Tape']],
['Rob Corddry', movies['Sex Tape']],
['Jason Segel', movies['Sex Tape']],
['Ellie Kemper', movies['Sex Tape']]
]
self.actors_played_in_bad_drama_or_comedy_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m:movie)
WHERE (m.genre = "Drama" OR m.genre = "Comedy")
AND m.rating < 5.5 AND m.votes > 50000
RETURN a.name, m
ORDER BY m.rating""",
description='Which actors played in bad drama or comedy?',
max_run_time_ms=4,
expected_result=expected_result)
##################################################################
### actors_played_in_drama_action_comedy
##################################################################
expected_result = [['Bradley Cooper'], ['Michael B. Jordan'],
['Michael Caine'], ['Miles Teller']]
self.actors_played_in_drama_action_comedy_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m0:movie {genre:'Action'}),
(a)-[:act]->(m1:movie {genre:'Drama'}),
(a)-[:act]->(m2:movie {genre:'Comedy'})
RETURN DISTINCT a.name""",
description=
'Which actors played in Action, Drama and Comedy movies?',
max_run_time_ms=1.5,
expected_result=expected_result)
##################################################################
### young_actors_played_with_cameron_diaz_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Nicolette Pierini'], 'Annie'],
[actors['Kate Upton'], 'The Other Woman']]
self.young_actors_played_with_cameron_diaz_query = QueryInfo(
query=
"""MATCH (Cameron:actor {name:"Cameron Diaz"})-[:act]->(m:movie)<-[:act]-(a:actor)
WHERE a.age < 35
RETURN a, m.title""",
description='Which young actors played along side Cameron Diaz?',
max_run_time_ms=5,
expected_result=expected_result)
##################################################################
### actors_played_with_cameron_diaz_and_younger_than_her_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Jason Segel'], 'Sex Tape'],
[actors['Ellie Kemper'], 'Sex Tape'],
[actors['Nicolette Pierini'], 'Annie'],
[actors['Rose Byrne'], 'Annie'],
[actors['Kate Upton'], 'The Other Woman'],
[actors['Nicki Minaj'], 'The Other Woman'],
[actors['Taylor Kinney'], 'The Other Woman']]
self.actors_played_with_cameron_diaz_and_younger_than_her_query = QueryInfo(
query=
"""MATCH (Cameron:actor {name:"Cameron Diaz"})-[:act]->(m:movie)<-[:act]-(a:actor)
WHERE a.age < Cameron.age
RETURN a, m.title order by a.name""",
description=
'Which actors played along side Cameron Diaz and are younger then her?',
max_run_time_ms=7,
expected_result=expected_result)
##################################################################
### sum_and_average_age_of_straight_outta_compton_cast_query
##################################################################
self.sum_and_average_age_of_straight_outta_compton_cast_query = QueryInfo(
query=
"""MATCH (a:actor)-[:act]->(m:movie{title:"Straight Outta Compton"})
RETURN m.title, SUM(a.age), AVG(a.age)""",
description=
'What is the sum and average age of the Straight Outta Compton cast?',
max_run_time_ms=4,
expected_result=[['Straight Outta Compton', 131, 32.75]])
##################################################################
### how_many_movies_cameron_diaz_played_query
##################################################################
self.how_many_movies_cameron_diaz_played_query = QueryInfo(
query="""MATCH (Cameron:actor{name:"Cameron Diaz"})-[:act]->(m:movie)
RETURN Cameron.name, COUNT(m.title)""",
description='In how many movies did Cameron Diaz played?',
max_run_time_ms=1.2,
expected_result=[['Cameron Diaz', 3]])
##################################################################
### find_ten_oldest_actors_query
##################################################################
expected_result = None
expected_result = None
if nodesAvailable:
expected_result = [[actors['Vincent Price']],
[actors['George Kennedy']],
[actors['Cloris Leachman']],
[actors['John Cullum']], [actors['Lois Smith']],
[actors['Robert Duvall']],
[actors['Olympia Dukakis']],
[actors['Ellen Burstyn']],
[actors['Michael Caine']],
[actors['Judi Dench']]]
self.find_ten_oldest_actors_query = QueryInfo(
query="""MATCH (a:actor)
RETURN DISTINCT *
ORDER BY a.age DESC
LIMIT 10""",
description='10 Oldest actors?',
max_run_time_ms=4.5,
expected_result=expected_result)
##################################################################
### actors_over_85_index_scan
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Michael Caine']],
[actors['Ellen Burstyn']],
[actors['Robert Duvall']],
[actors['Olympia Dukakis']],
[actors['Lois Smith']], [actors['John Cullum']],
[actors['Cloris Leachman']],
[actors['George Kennedy']],
[actors['Vincent Price']]]
self.actors_over_85_index_scan = QueryInfo(
query="""MATCH (a:actor)
WHERE a.age > 85
RETURN *
ORDER BY a.age, a.name""",
description='Actors over 85 on indexed property?',
max_run_time_ms=1.5,
expected_result=expected_result)
##################################################################
### eighties_movies_index_scan
##################################################################
self.eighties_movies_index_scan = QueryInfo(
query="""MATCH (m:movie)
WHERE 1980 <= m.year
AND m.year < 1990
RETURN m.title, m.year
ORDER BY m.year""",
description='Multiple filters on indexed property?',
max_run_time_ms=1.5,
expected_result=[['The Evil Dead', 1981], ['Vincent', 1982]])
##################################################################
### find_titles_starting_with_american_query
##################################################################
self.find_titles_starting_with_american_query = QueryInfo(
query="""MATCH (m:movie)
WHERE LEFT(m.title, 8) = 'American'
RETURN m.title
ORDER BY m.title""",
description='Movies starting with "American"?',
max_run_time_ms=4,
expected_result=[['American Honey'], ['American Pastoral'],
['American Sniper']])
##################################################################
### all_actors_named_tim
##################################################################
# self.all_actors_named_tim = QueryInfo(
# query="""CALL db.idx.fulltext.queryNodes('actor', 'tim')""",
# description='All actors named Tim',
# max_run_time_ms=4,
# expected_result=[['Tim Roth', 0],
# ['Tim Reid', 0],
# ['Tim McGraw', 0],
# ['Tim Griffin', 0],
# ['Tim Blake Nelson', 0]]
# )
self.queries_info = [
self.number_of_actors_query,
self.actors_played_with_nicolas_cage_query,
self.find_three_actors_played_with_nicolas_cage_query,
self.actors_played_in_movie_straight_outta_compton_query,
self.actors_over_50_that_played_in_blockbusters_query,
self.actors_played_in_bad_drama_or_comedy_query,
self.actors_played_in_drama_action_comedy_query,
self.young_actors_played_with_cameron_diaz_query,
self.actors_played_with_cameron_diaz_and_younger_than_her_query,
self.sum_and_average_age_of_straight_outta_compton_cast_query,
self.how_many_movies_cameron_diaz_played_query,
self.find_ten_oldest_actors_query, self.actors_over_85_index_scan,
self.eighties_movies_index_scan,
self.find_titles_starting_with_american_query
# self.all_actors_named_tim
]
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
my_friends_query = QueryInfo(
query="""MATCH (ME:person {name:"Roi Lipman"})-[:friend]->(f:person)
RETURN f.name""",
description='My friends?',
max_run_time_ms=0.25,
expected_result=[['Tal Doron'],
['Omri Traub'],
['Boaz Arad'],
['Ori Laslo'],
['Ailon Velger'],
['Alon Fital']]
)
friends_of_friends_query = QueryInfo(
query="""MATCH (ME:person {name:"Roi Lipman"})-[:friend]->(:person)-[:friend]->(fof:person)
RETURN fof.name""",
description='Friends of friends?',
max_run_time_ms=0.25,
expected_result=[['Valerie Abigail Arad'],
['Shelly Laslo Rooz'],
['Noam Nativ'],
['Jane Chernomorin'],
['Mor Yesharim'],
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (n:actor{name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title""",
description='Which actors played along side Nicolas Cage?',
max_run_time_ms=3,
expected_result=[['"Cassi Thomson"', '"Left Behind"'],
['"Gary Grubbs"', '"Left Behind"'],
['"Quinton Aaron"', '"Left Behind"'],
['"Martin Klebba"', '"Left Behind"'],
['"Lea Thompson"', '"Left Behind"'],
['"Nicolas Cage"', '"Left Behind"'],
['"Chad Michael Murray"', '"Left Behind"'],
['"Jordin Sparks"', '"Left Behind"']])
find_three_actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (nicolas:actor {name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title
LIMIT 3""",
description='Get 3 actors who have played along side Nicolas Cage?',
max_run_time_ms=2.5,
expected_result=[['"Cassi Thomson"', '"Left Behind"'],
['"Gary Grubbs"', '"Left Behind"'],
['"Quinton Aaron"', '"Left Behind"'],
graph_entities = QueryInfo(
query="""MATCH (e) RETURN e.name, LABELS(e) as label ORDER BY label, e.name""",
description='Returns each node in the graph, specifing node label.',
max_run_time_ms=0.2,
expected_result=[['Netherlands','country'],
['Andora','country'],
['Canada','country'],
['China','country'],
['Germany','country'],
['Greece','country'],
['Italy','country'],
['Japan','country'],
['Kazakhstan','country'],
['Prague','country'],
['Russia','country'],
['Thailand','country'],
['USA','country'],
['Ailon Velger','person'],
['Alon Fital','person'],
['Boaz Arad','person'],
['Gal Derriere','person'],
['Jane Chernomorin','person'],
['Lucy Yanfital','person'],
['Mor Yesharim','person'],
['Noam Nativ','person'],
['Omri Traub','person'],
['Ori Laslo','person'],
['Roi Lipman','person'],
['Shelly Laslo Rooz','person'],
['Tal Doron','person'],
['Valerie Abigail Arad','person']]
)
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
graph_entities = QueryInfo(
query=
"""MATCH (e) RETURN e.name, LABELS(e) as label ORDER BY label, e.name""",
description='Returns each node in the graph, specifing node label.',
max_run_time_ms=0.2,
expected_result=[['Amsterdam', 'country'], ['Andora', 'country'],
['Canada', 'country'], ['China', 'country'],
['Germany', 'country'], ['Greece', 'country'],
['Italy', 'country'], ['Japan', 'country'],
['Kazakhstan', 'country'], ['Prague', 'country'],
['Russia', 'country'], ['Thailand', 'country'],
['USA', 'country'], ['Ailon Velger', 'person'],
['Alon Fital', 'person'], ['Boaz Arad', 'person'],
['Gal Derriere',
'person'], ['Jane Chernomorin', 'person'],
['Lucy Yanfital', 'person'], ['Mor Yesharim', 'person'],
['Noam Nativ', 'person'], ['Omri Traub', 'person'],
['Ori Laslo', 'person'], ['Roi Lipman', 'person'],
['Shelly Laslo Rooz', 'person'], ['Tal Doron', 'person'],
['Valerie Abigail Arad', 'person']])
relation_type_counts = QueryInfo(
query=
"""MATCH ()-[e]->() RETURN TYPE(e) as relation_type, COUNT(e) as num_relations ORDER BY relation_type, num_relations""",
description='Returns each relation type in the graph and its count.',
max_run_time_ms=0.4,
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (n:actor{name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title""",
description='Which actors played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Quinton Aaron', 'Left Behind'],
['Martin Klebba', 'Left Behind'],
['Lea Thompson', 'Left Behind'],
['Nicolas Cage', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Jordin Sparks', 'Left Behind']])
find_three_actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (nicolas:actor {name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title
LIMIT 3""",
description='Get 3 actors who have played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Quinton Aaron', 'Left Behind'],
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../../')
from demo import QueryInfo
actors_played_with_nicolas_cage_query = QueryInfo(
query="""MATCH (n:actor{name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title""",
description='Which actors played along side Nicolas Cage?',
max_run_time_ms=5,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Quinton Aaron', 'Left Behind'],
['Martin Klebba', 'Left Behind'],
['Lea Thompson', 'Left Behind'],
['Nicolas Cage', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Jordin Sparks', 'Left Behind']]
)
find_three_actors_played_with_nicolas_cage_query = QueryInfo(
query="""MATCH (nicolas:actor {name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title
LIMIT 3""",
description='Get 3 actors who have played along side Nicolas Cage?',
max_run_time_ms=2.5,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
def __init__(self, actors=None, movies=None):
nodesAvailable = (movies is not None and actors is not None)
##################################################################
### number_of_actors_query
##################################################################
self.number_of_actors_query = QueryInfo(
query="""MATCH (n:actor) RETURN count(n) as actors_count""",
description='How many actors are in the graph?',
max_run_time_ms=0.2,
expected_result=[[1317]])
##################################################################
### actors_played_with_nicolas_cage_query
##################################################################
self.actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (n:actor{name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title ORDER BY a.name""",
description='Which actors played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Gary Grubbs', 'Left Behind'],
['Jordin Sparks', 'Left Behind'],
['Lea Thompson', 'Left Behind'],
['Martin Klebba', 'Left Behind'],
['Nicolas Cage', 'Left Behind'],
['Quinton Aaron', 'Left Behind']])
##################################################################
### find_three_actors_played_with_nicolas_cage_query
##################################################################
self.find_three_actors_played_with_nicolas_cage_query = QueryInfo(
query=
"""MATCH (nicolas:actor {name:"Nicolas Cage"})-[:act]->(m:movie)<-[:act]-(a:actor)
RETURN a.name, m.title
ORDER BY a.name, m.title
LIMIT 3""",
description='Get 3 actors who have played along side Nicolas Cage?',
max_run_time_ms=4,
expected_result=[['Cassi Thomson', 'Left Behind'],
['Chad Michael Murray', 'Left Behind'],
['Gary Grubbs', 'Left Behind']])
##################################################################
### actors_played_in_movie_straight_outta_compton_query
##################################################################
self.actors_played_in_movie_straight_outta_compton_query = QueryInfo(
query=
"""MATCH (a:actor)-[:act]->(m:movie {title:"Straight Outta Compton"})
RETURN a.name""",
description=
'Which actors played in the movie Straight Outta Compton?',
max_run_time_ms=3.5,
expected_result=[['Aldis Hodge'], ['Corey Hawkins'],
['Neil Brown Jr.'], ['O\'Shea Jackson Jr.']])
##################################################################
### actors_over_50_that_played_in_blockbusters_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [
[actors['Bill Irwin'], movies['Interstellar']],
[actors['Vincent Price'], movies['Vincent']],
[actors['Ellen Burstyn'], movies['Interstellar']],
[actors['Paul Reiser'], movies['Whiplash']],
[actors['Francis X. McCarthy'], movies['Interstellar']],
[actors['John Lithgow'], movies['Interstellar']],
[actors['J.K. Simmons'], movies['Whiplash']],
[actors['Chris Mulkey'], movies['Whiplash']],
[actors['Rachael Harris'], movies['Lucifer']],
[actors['Matthew McConaughey'], movies['Interstellar']],
[actors['D.B. Woodside'], movies['Lucifer']]
]
self.actors_over_50_that_played_in_blockbusters_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m:movie)
WHERE a.age >= 50 AND m.votes > 10000 AND m.rating > 8.2
RETURN a, m ORDER BY a.name, m.name""",
description=
'Which actors who are over 50 played in blockbuster movies?',
max_run_time_ms=4.0,
expected_result=expected_result)
##################################################################
### actors_played_in_bad_drama_or_comedy_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [
['Rita Ora', movies['Fifty Shades of Grey']],
['Dakota Johnson', movies['Fifty Shades of Grey']],
['Marcia Gay Harden', movies['Fifty Shades of Grey']],
['Jamie Dornan', movies['Fifty Shades of Grey']],
['Eloise Mumford', movies['Fifty Shades of Grey']],
['Max Martini', movies['Fifty Shades of Grey']],
['Luke Grimes', movies['Fifty Shades of Grey']],
['Jennifer Ehle', movies['Fifty Shades of Grey']],
['Victor Rasuk', movies['Fifty Shades of Grey']],
['Nancy Lenehan', movies['Sex Tape']],
['Rob Lowe', movies['Sex Tape']],
['Cameron Diaz', movies['Sex Tape']],
['Rob Corddry', movies['Sex Tape']],
['Jason Segel', movies['Sex Tape']],
['Ellie Kemper', movies['Sex Tape']]
]
self.actors_played_in_bad_drama_or_comedy_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m:movie)
WHERE (m.genre = "Drama" OR m.genre = "Comedy")
AND m.rating < 5.5 AND m.votes > 50000
RETURN a.name, m
ORDER BY m.rating""",
description='Which actors played in bad drama or comedy?',
max_run_time_ms=4,
expected_result=expected_result)
##################################################################
### actors_played_in_drama_action_comedy
##################################################################
expected_result = [['Bradley Cooper'], ['Michael B. Jordan'],
['Michael Caine'], ['Miles Teller']]
self.actors_played_in_drama_action_comedy_query = QueryInfo(
query="""MATCH (a:actor)-[:act]->(m0:movie {genre:'Action'}),
(a)-[:act]->(m1:movie {genre:'Drama'}),
(a)-[:act]->(m2:movie {genre:'Comedy'})
RETURN DISTINCT a.name""",
description=
'Which actors played in Action, Drama and Comedy movies?',
reversible=False,
max_run_time_ms=1.5,
expected_result=expected_result)
##################################################################
### young_actors_played_with_cameron_diaz_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Nicolette Pierini'], 'Annie'],
[actors['Kate Upton'], 'The Other Woman']]
self.young_actors_played_with_cameron_diaz_query = QueryInfo(
query=
"""MATCH (Cameron:actor {name:"Cameron Diaz"})-[:act]->(m:movie)<-[:act]-(a:actor)
WHERE a.age < 35
RETURN a, m.title ORDER BY m.title""",
description='Which young actors played along side Cameron Diaz?',
max_run_time_ms=5,
expected_result=expected_result)
##################################################################
### actors_played_with_cameron_diaz_and_younger_than_her_query
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Jason Segel'], 'Sex Tape'],
[actors['Ellie Kemper'], 'Sex Tape'],
[actors['Nicolette Pierini'], 'Annie'],
[actors['Rose Byrne'], 'Annie'],
[actors['Kate Upton'], 'The Other Woman'],
[actors['Nicki Minaj'], 'The Other Woman'],
[actors['Taylor Kinney'], 'The Other Woman']]
self.actors_played_with_cameron_diaz_and_younger_than_her_query = QueryInfo(
query=
"""MATCH (Cameron:actor {name:"Cameron Diaz"})-[:act]->(m:movie)<-[:act]-(a:actor)
WHERE a.age < Cameron.age
RETURN a, m.title order by a.name""",
description=
'Which actors played along side Cameron Diaz and are younger then her?',
max_run_time_ms=7,
expected_result=expected_result)
##################################################################
### sum_and_average_age_of_straight_outta_compton_cast_query
##################################################################
self.sum_and_average_age_of_straight_outta_compton_cast_query = QueryInfo(
query=
"""MATCH (a:actor)-[:act]->(m:movie{title:"Straight Outta Compton"})
RETURN m.title, SUM(a.age), AVG(a.age)""",
description=
'What is the sum and average age of the Straight Outta Compton cast?',
max_run_time_ms=4,
expected_result=[['Straight Outta Compton', 131, 32.75]])
##################################################################
### how_many_movies_cameron_diaz_played_query
##################################################################
self.how_many_movies_cameron_diaz_played_query = QueryInfo(
query="""MATCH (Cameron:actor{name:"Cameron Diaz"})-[:act]->(m:movie)
RETURN Cameron.name, COUNT(m.title)""",
description='In how many movies did Cameron Diaz played?',
max_run_time_ms=1.2,
expected_result=[['Cameron Diaz', 3]])
##################################################################
### find_ten_oldest_actors_query
##################################################################
expected_result = None
expected_result = None
if nodesAvailable:
expected_result = [[actors['Vincent Price']],
[actors['George Kennedy']],
[actors['Cloris Leachman']],
[actors['John Cullum']], [actors['Lois Smith']],
[actors['Robert Duvall']],
[actors['Olympia Dukakis']],
[actors['Ellen Burstyn']],
[actors['Michael Caine']],
[actors['Judi Dench']]]
self.find_ten_oldest_actors_query = QueryInfo(
query="""MATCH (a:actor)
RETURN DISTINCT *
ORDER BY a.age DESC
LIMIT 10""",
description='10 Oldest actors?',
max_run_time_ms=4.5,
expected_result=expected_result)
##################################################################
### actors_over_85_index_scan
##################################################################
expected_result = None
if nodesAvailable:
expected_result = [[actors['Michael Caine']],
[actors['Ellen Burstyn']],
[actors['Robert Duvall']],
[actors['Olympia Dukakis']],
[actors['Lois Smith']], [actors['John Cullum']],
[actors['Cloris Leachman']],
[actors['George Kennedy']],
[actors['Vincent Price']]]
self.actors_over_85_index_scan = QueryInfo(
query="""MATCH (a:actor)
WHERE a.age > 85
RETURN *
ORDER BY a.age, a.name""",
description='Actors over 85 on indexed property?',
max_run_time_ms=1.5,
expected_result=expected_result)
##################################################################
### eighties_movies_index_scan
##################################################################
self.eighties_movies_index_scan = QueryInfo(
query="""MATCH (m:movie)
WHERE 1980 <= m.year
AND m.year < 1990
RETURN m.title, m.year
ORDER BY m.year""",
description='Multiple filters on indexed property?',
max_run_time_ms=1.5,
expected_result=[['The Evil Dead', 1981], ['Vincent', 1982]])
##################################################################
### find_titles_starting_with_american_query
##################################################################
self.find_titles_starting_with_american_query = QueryInfo(
query="""MATCH (m:movie)
WHERE LEFT(m.title, 8) = 'American'
RETURN m.title
ORDER BY m.title""",
description='Movies starting with "American"?',
max_run_time_ms=4,
expected_result=[['American Honey'], ['American Pastoral'],
['American Sniper']])
##################################################################
### same_year_higher_rating_than_huntforthewilderpeople
##################################################################
self.same_year_higher_rating_than_huntforthewilderpeople_query = QueryInfo(
query="""MATCH (base:movie), (option:movie)
WHERE base.title = 'Hunt for the Wilderpeople' AND
base.year = option.year AND
option.rating > base.rating
RETURN option.title, option.rating
ORDER BY option.rating, option.title desc
LIMIT 10""",
description=
'List 10 movies released on the same year as "Hunt for the Wilderpeople" that got higher rating than it',
reversible=False,
max_run_time_ms=0.8,
expected_result=[["Hacksaw Ridge", 8.8], ["Moonlight", 8.7],
["La La Land", 8.6], ["Arrival", 8.5],
["Hell or High Water", 8.2], ["Zootopia", 8.1],
["Split", 8.1], ["Nocturnal Animals", 8.1],
["Deadpool", 8.1], ["Ah-ga-ssi", 8.1]])
##################################################################
### all_actors_named_tim
##################################################################
self.all_actors_named_tim = QueryInfo(
query="""CALL db.idx.fulltext.queryNodes('actor', 'tim') YIELD node
RETURN node.name
ORDER BY node.name""",
description='All actors named Tim',
reversible=False,
max_run_time_ms=4,
expected_result=[['Tim Roth'], ['Tim Reid'], ['Tim McGraw'],
['Tim Griffin'], ['Tim Blake Nelson']])
##################################################################
### grand_budapest_hotel_cast_and_their_other_roles
##################################################################
self.grand_budapest_hotel_cast_and_their_other_roles = QueryInfo(
query=
"""MATCH (a:actor)-[:act]->(h:movie {title: 'The Grand Budapest Hotel'})
OPTIONAL MATCH (a)-[:act]->(m:movie) WHERE m <> h
RETURN a.name, m.title
ORDER BY a.name, m.title""",
description=
'All actors in The Grand Budapest Hotel and their other movies',
reversible=False,
max_run_time_ms=4,
expected_result=[['Adrien Brody', None],
['Bill Murray', 'The Jungle Book'],
['F. Murray Abraham', None],
['Harvey Keitel', 'The Ridiculous 6'],
['Harvey Keitel', 'Youth'],
['Jeff Goldblum', 'Independence Day: Resurgence'],
['Jude Law', 'Spy'], ['Mathieu Amalric', None],
['Ralph Fiennes', 'A Bigger Splash'],
['Ralph Fiennes', 'Spectre'],
['Willem Dafoe', 'John Wick'],
['Willem Dafoe', 'The Fault in Our Stars']])
self.queries_info = [
self.number_of_actors_query,
self.actors_played_with_nicolas_cage_query,
self.find_three_actors_played_with_nicolas_cage_query,
self.actors_played_in_movie_straight_outta_compton_query,
self.actors_over_50_that_played_in_blockbusters_query,
self.actors_played_in_bad_drama_or_comedy_query,
self.actors_played_in_drama_action_comedy_query,
self.young_actors_played_with_cameron_diaz_query,
self.actors_played_with_cameron_diaz_and_younger_than_her_query,
self.sum_and_average_age_of_straight_outta_compton_cast_query,
self.how_many_movies_cameron_diaz_played_query,
self.find_ten_oldest_actors_query, self.actors_over_85_index_scan,
self.eighties_movies_index_scan,
self.find_titles_starting_with_american_query,
self.same_year_higher_rating_than_huntforthewilderpeople_query,
self.all_actors_named_tim,
self.grand_budapest_hotel_cast_and_their_other_roles
]
