def test_get_upstream():
m = MRA()
egfr = Agent('EGFR', db_refs={'HGNC': '3236', 'TEXT': 'EGFR'})
kras = Agent('KRAS', db_refs={'HGNC': '6407', 'TEXT': 'KRAS'})
stmts = [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')
mm = MRA_Module(testing=True)
mm.mra = m
kras_term = ekb_from_agent(kras)
msg = KQMLList('MODEL-GET-UPSTREAM')
msg.sets('target', kras_term)
msg.set('model-id', str(model_id))
print(msg)
reply = mm.respond_model_get_upstream(msg)
ups = reply.get('upstream')
assert (len(ups) == 1)
print(reply)
