from indra.sources import rlimsp


def test_simple_usage():
    rp = rlimsp.process_pmc('PMC3717945')
    stmts = rp.statements
    assert len(stmts) == 6, len(stmts)
    for s in stmts:
        assert len(s.evidence) == 1, "Wrong amount of evidence."
        ev = s.evidence[0]
        assert ev.annotations, "Missing annotations."
        assert 'agents' in ev.annotations.keys()
        assert 'trigger' in ev.annotations.keys()


def test_ungrounded_usage():
    rp = rlimsp.process_pmc('PMC3717945', with_grounding=False)
    assert len(rp.statements) == 33, len(rp.statements)

from indra.sources import rlimsp


def test_simple_usage():
    rp = rlimsp.process_from_webservice('PMC3717945')
    stmts = rp.statements
    assert len(stmts) == 6, len(stmts)
    for s in stmts:
        assert len(s.evidence) == 1, "Wrong amount of evidence."
        ev = s.evidence[0]
        assert ev.annotations, "Missing annotations."
        assert 'agents' in ev.annotations.keys()
        assert 'trigger' in ev.annotations.keys()


def test_ungrounded_usage():
    rp = rlimsp.process_from_webservice('PMC3717945', with_grounding=False)
    assert len(rp.statements) == 33, len(rp.statements)


def test_grounded_endpoint_with_pmids():
    pmid_list = ['16403219', '22258404', '16961925', '22096607']
    for pmid in pmid_list:
        rp = rlimsp.process_from_webservice(pmid, id_type='pmid',
                                            with_grounding=False)
        assert len(rp.statements) > 10, len(rp.statements)

Add a test for pmids and update to new api.
Add a test for pmids and update to new api.

  from indra.sources import rlimsp
  
  
  def test_simple_usage():
-     rp = rlimsp.process_pmc('PMC3717945')
+     rp = rlimsp.process_from_webservice('PMC3717945')
      stmts = rp.statements
      assert len(stmts) == 6, len(stmts)
      for s in stmts:
          assert len(s.evidence) == 1, "Wrong amount of evidence."
          ev = s.evidence[0]
          assert ev.annotations, "Missing annotations."
          assert 'agents' in ev.annotations.keys()
          assert 'trigger' in ev.annotations.keys()
  
  
  def test_ungrounded_usage():
-     rp = rlimsp.process_pmc('PMC3717945', with_grounding=False)
+     rp = rlimsp.process_from_webservice('PMC3717945', with_grounding=False)
      assert len(rp.statements) == 33, len(rp.statements)
  
+ 
+ def test_grounded_endpoint_with_pmids():
+     pmid_list = ['16403219', '22258404', '16961925', '22096607']
+     for pmid in pmid_list:
+         rp = rlimsp.process_from_webservice(pmid, id_type='pmid',
+                                             with_grounding=False)
+         assert len(rp.statements) > 10, len(rp.statements)
+ 
Add a test for pmids and update to new api.
## Code Before:
from indra.sources import rlimsp


def test_simple_usage():
    rp = rlimsp.process_pmc('PMC3717945')
    stmts = rp.statements
    assert len(stmts) == 6, len(stmts)
    for s in stmts:
        assert len(s.evidence) == 1, "Wrong amount of evidence."
        ev = s.evidence[0]
        assert ev.annotations, "Missing annotations."
        assert 'agents' in ev.annotations.keys()
        assert 'trigger' in ev.annotations.keys()


def test_ungrounded_usage():
    rp = rlimsp.process_pmc('PMC3717945', with_grounding=False)
    assert len(rp.statements) == 33, len(rp.statements)

## Instruction:
Add a test for pmids and update to new api.
## Code After:
from indra.sources import rlimsp


def test_simple_usage():
    rp = rlimsp.process_from_webservice('PMC3717945')
    stmts = rp.statements
    assert len(stmts) == 6, len(stmts)
    for s in stmts:
        assert len(s.evidence) == 1, "Wrong amount of evidence."
        ev = s.evidence[0]
        assert ev.annotations, "Missing annotations."
        assert 'agents' in ev.annotations.keys()
        assert 'trigger' in ev.annotations.keys()


def test_ungrounded_usage():
    rp = rlimsp.process_from_webservice('PMC3717945', with_grounding=False)
    assert len(rp.statements) == 33, len(rp.statements)


def test_grounded_endpoint_with_pmids():
    pmid_list = ['16403219', '22258404', '16961925', '22096607']
    for pmid in pmid_list:
        rp = rlimsp.process_from_webservice(pmid, id_type='pmid',
                                            with_grounding=False)
        assert len(rp.statements) > 10, len(rp.statements)
