def test_get_string():
st = [Phosphorylation(Agent('MAP2K1'), Agent('MAPK1'))]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
graph_str = ga.get_string()
assert (graph_str)
def test_complex():
st = [Complex([Agent('BRAF'), Agent('RAF1'), Agent('YWAH')])]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
assert (len(ga.graph.nodes()) == 3)
assert (len(ga.graph.edges()) == 3)
def test_phosphorylation():
st = [Phosphorylation(Agent('MAP2K1'), Agent('MAPK1'))]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
assert (len(ga.graph.nodes()) == 2)
assert (len(ga.graph.edges()) == 1)
def test_inactivation():
st = [Inhibition(Agent('DUSP4'), Agent('MAPK1'))]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
assert (len(ga.graph.nodes()) == 2)
assert (len(ga.graph.edges()) == 1)
def test_dephosphorylation_noenz():
st = [Dephosphorylation(None, Agent('MAPK1'))]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
assert (len(ga.graph.nodes()) == 0)
assert (len(ga.graph.edges()) == 0)
def test_save_pdf():
st = [Phosphorylation(Agent('MAP2K1'), Agent('MAPK1'))]
ga = GraphAssembler()
ga.add_statements(st)
ga.make_model()
ga.save_pdf('/dev/null')
