def test_get_edges(clean_slate, query):
g = get_graph('kgx-unit-test')[3]
t = NeoTransformer(g,
uri=DEFAULT_NEO4J_URL,
username=DEFAULT_NEO4J_USERNAME,
password=DEFAULT_NEO4J_PASSWORD)
t.save()
for k, v in query[1].items():
t.set_edge_filter(k, v)
edges = t.get_edges()
edge_list = [x[1] for x in edges]
assert len(edges) == query[2]
help='A filter that can be applied to nodes')
parser.add_argument('--edge_filter',
action='append',
help='A filter that can be applied to edges')
parser.add_argument('--uri',
help='URI/URL for Neo4j (including port)',
default='localhost:7474')
parser.add_argument('--username', help='username', default='neo4j')
parser.add_argument('--password', help='password', default='demo')
args = parser.parse_args()
# Initialize NeoTransformer
nt = NeoTransformer(None,
uri=args.uri,
username=args.username,
password=args.password)
if args.node_filter:
for f in args.node_filter:
k, v = f.split('=')
nt.set_node_filter(k, set(v))
if args.edge_filter:
for f in args.edge_filter:
k, v = f.split('=')
nt.set_edge_filter(k, set(v))
# Read from Neo4j with the given filter constraints (if any)
nt.load()
nt.report()
