def test_model_undo():
m = MRA()
stmts1 = [sts.Phosphorylation(sts.Agent('A'), sts.Agent('B'))]
m.new_model(stmts1)
res = m.model_undo()
action = res.get('action')
assert action is not None
assert action.get('action') == 'remove_stmts'
assert action.get('statements') == stmts1
def test_transformations():
m = MRA()
stmts1 = [sts.Phosphorylation(sts.Agent('A'), sts.Agent('B'))]
m.new_model(stmts1)
assert(len(m.transformations) == 1)
tr = m.transformations[0]
assert(tr[0] == 'add_stmts')
assert(tr[1] == stmts1)
assert(tr[2] is None)
assert(tr[3] == 1)
stmts2 = [sts.Phosphorylation(sts.Agent('C'), sts.Agent('D'))]
m.extend_model(stmts2, 1)
assert(len(m.transformations) == 2)
tr = m.transformations[1]
assert(tr[0] == 'add_stmts')
assert(tr[1] == stmts2)
assert(tr[2] == 1)
assert(tr[3] == 2)
def test_get_upstream():
m = MRA()
egfr = sts.Agent('EGFR', db_refs={'HGNC': '3236', 'TEXT': 'EGFR'})
kras = sts.Agent('KRAS', db_refs={'HGNC': '6407', 'TEXT': 'KRAS'})
stmts = [sts.Activation(egfr, kras)]
model_id = m.new_model(stmts)
upstream = m.get_upstream(kras, model_id)
assert(len(upstream) == 1)
assert(upstream[0].name == 'EGFR')
