import os
from gremthon import Gremthon
example_graph_son = '{}/doctors-consumers-graph.json'.format(os.environ['TEST_DRIVE_DIR'])
g = Gremthon(rexster.getGraph("graph"))
count = g.V.count()
if count == 0:
    g.input_graph(example_graph_son)
def test_input_graph():
    temp_titan_graph = TitanFactory.build().set("storage.backend", "inmemory").open()
    temp_g = Gremthon(temp_titan_graph)
    temp_g.input_graph(graph_son_filename)
    assert temp_g.V.count() == 2
def test_input_graph():
    temp_titan_graph = TitanFactory.build().set('storage.backend','inmemory').open()
    temp_g = Gremthon(temp_titan_graph)
    temp_g.input_graph(graph_son_filename)
    assert temp_g.V.count() == 2