def test_get_statements_by_hash():
hash_list = [30674674032092136, -22289282229858243, -25056605420392180]
stmts = dbr.get_statements_by_hash(hash_list, simple_response=True)
print({s.get_hash(shallow=True): s for s in stmts})
assert len(stmts) >= 2, \
"Wrong number of statements: %s vs. %s" % (len(stmts), len(hash_list))
p = dbr.get_statements_by_hash(hash_list)
assert len(p.statements)
assert len(p.get_source_counts())
assert len(p.get_ev_counts())
return
def test_get_statements_by_hash():
hash_list = [-36028793042562873, -12978096432588272, -12724735151233845]
stmts = dbr.get_statements_by_hash(hash_list)
print({s.get_hash(shallow=True): s for s in stmts})
assert len(stmts) == len(hash_list), \
"Wrong number of statements: %s vs. %s" % (len(stmts), len(hash_list))
return
def test_get_statements_by_hash():
hash_list = [-36028793042562873, -12978096432588272, -12724735151233845]
stmts = dbr.get_statements_by_hash(hash_list)
print({s.get_hash(shallow=True): s for s in stmts})
assert len(stmts) == len(hash_list), \
"Wrong number of statements: %s vs. %s" % (len(stmts), len(hash_list))
return
def download_statements(hashes):
"""Download the INDRA Statements corresponding to a set of hashes.
"""
stmts_by_hash = {}
for group in tqdm.tqdm(batch_iter(hashes, 200),
total=int(len(hashes) / 200)):
idbp = indra_db_rest.get_statements_by_hash(list(group), ev_limit=10)
for stmt in idbp.statements:
stmts_by_hash[stmt.get_hash()] = stmt
return stmts_by_hash
def download_statements(df, ev_limit=5):
"""Download the INDRA Statements corresponding to entries in a data frame.
"""
all_stmts = []
for idx, group in enumerate(batch_iter(df.hash, 500)):
logger.info('Getting statement batch %d' % idx)
idbp = indra_db_rest.get_statements_by_hash(list(group),
ev_limit=ev_limit)
all_stmts += idbp.statements
return all_stmts
def download_statements(df, beliefs, ev_limit=5):
"""Download the INDRA Statements corresponding to entries in a data frame.
"""
all_stmts = []
unique_hashes = list(set(df.stmt_hash))
batches = list(batch_iter(unique_hashes, 500))
logger.info('Getting %d unique hashes from db' % len(unique_hashes))
for group in tqdm.tqdm(batches):
idbp = indra_db_rest.get_statements_by_hash(list(group),
ev_limit=ev_limit)
all_stmts += idbp.statements
for stmt in all_stmts:
belief = beliefs.get(stmt.get_hash())
if belief is None:
logger.info('No belief found for %s' % str(stmt))
continue
stmt.belief = belief
return all_stmts
def test_get_statements_by_hash_no_hash():
p = dbr.get_statements_by_hash([])
assert not p.statements, "Got statements without giving a hash."
def test_get_statements_by_hash_no_hash():
stmts = dbr.get_statements_by_hash([])
assert not stmts, "Got statements without giving a hash."
def test_get_statements_by_hash_no_hash():
stmts = dbr.get_statements_by_hash([], simple_response=True)
assert not stmts, "Got statements without giving a hash."
