def test_cypher_full(self):
sio = io.StringIO()
exporter = CypherGraphExporter(sio, edge_types=edge_types())
for entity in ENTITIES:
proxy = model.get_proxy(entity)
exporter.write(proxy)
value = sio.getvalue()
assert 'entity:company' in value, value
assert 'tel:+12025557612' in value, value
def test_csv_export(self):
exporter = Neo4JCSVExporter(self.outdir,
extra=['source'],
edge_types=edge_types())
for entity in ENTITIES:
entity = model.get_proxy(entity)
exporter.write(entity, extra=['test'])
fh, writer = exporter.handles[entity.schema]
outfile = fh.name
exporter.finalize()
fh = open(outfile, 'r')
csv_reader = csv.reader(fh)
rows = list(csv_reader)
headers = rows[0]
assert ':TYPE' in headers, headers
assert ':START_ID' in headers, headers
assert ':END_ID' in headers, headers
assert 'id' in headers, headers
assert 'date' in headers, headers
data = rows[1]
assert 'OWNERSHIP' in data, data
assert '2003-04-01' in data, data
def test_csv_export(self):
exporter = Neo4JCSVExporter(self.outdir,
extra=["source"],
edge_types=edge_types())
for entity in ENTITIES:
entity = model.get_proxy(entity)
exporter.write(entity, extra=["test"])
fh, writer = exporter.handles[entity.schema]
outfile = fh.name
exporter.finalize()
fh = open(outfile, "r")
csv_reader = csv.reader(fh)
rows = list(csv_reader)
headers = rows[0]
assert ":TYPE" in headers, headers
assert ":START_ID" in headers, headers
assert ":END_ID" in headers, headers
assert "id" in headers, headers
assert "date" in headers, headers
data = rows[1]
assert "OWNERSHIP" in data, data
assert "2003-04-01" in data, data
@click.option('-o', '--outfile', type=click.File('w'), default='-') # noqa
@click.option('--qualified/--unqualified',
is_flag=True,
default=True,
help='Generate full predicates') # noqa
def export_rdf(infile, outfile, qualified=True):
exporter = RDFExporter(outfile, qualified=qualified)
export_stream(exporter, infile)
@cli.command('export-gexf', help="Export to GEXF (Gephi) format")
@click.option('-i', '--infile', type=click.File('r'), default='-') # noqa
@click.option('-o', '--outfile', type=click.File('w'), default='-') # noqa
@click.option('-e',
'--edge-types',
type=click.Choice(edge_types()),
multiple=True,
default=DEFAULT_EDGE_TYPES,
help="Property types to be reified into graph edges.")
def export_gexf(infile, outfile, edge_types):
exporter = NXGraphExporter(outfile, edge_types=edge_types)
export_stream(exporter, infile)
@cli.command('export-cypher', help="Export to Cypher script")
@click.option('-i', '--infile', type=click.File('r'), default='-') # noqa
@click.option('-o', '--outfile', type=click.File('w'), default='-') # noqa
@click.option('-e',
'--edge-types',
type=click.Choice(edge_types()),
multiple=True,