def test_protein_to_transcript_cache(accession, version, match_version,
expected_forward, expected_reverse):
"""
Get transcript for protein and check the resulting cache state.
"""
try:
ncbi.protein_to_transcript(accession, version, match_version)
except ncbi.NoLinkError:
pass
forward = [(key.split(':')[-1], redis.get(key) or None)
for key in redis.keys('ncbi:transcript-to-protein:*')]
assert sorted(forward) == sorted(expected_forward)
reverse = [(redis.get(key) or None, key.split(':')[-1])
for key in redis.keys('ncbi:protein-to-transcript:*')]
assert sorted(reverse) == sorted(expected_reverse)
def test_protein_to_transcript_cache(accession, version, match_version,
expected_forward, expected_reverse):
"""
Get transcript for protein and check the resulting cache state.
"""
try:
ncbi.protein_to_transcript(accession, version, match_version)
except ncbi.NoLinkError:
pass
forward = [(key.split(':')[-1], redis.get(key) or None)
for key in redis.keys('ncbi:transcript-to-protein:*')]
assert sorted(forward) == sorted(expected_forward)
reverse = [(redis.get(key) or None, key.split(':')[-1])
for key in redis.keys('ncbi:protein-to-transcript:*')]
assert sorted(reverse) == sorted(expected_reverse)
def get_totals():
"""
Get the total for all known counters.
Known counters are just those that have been incremented at least once.
"""
counters = client.keys('counter:*:total')
pipe = client.pipeline(transaction=False)
for counter in counters:
pipe.get(counter)
return {counter.split(':')[1]: int(value)
for counter, value in zip(counters, pipe.execute())}
def get_totals():
"""
Get the total for all known counters.
Known counters are just those that have been incremented at least once.
"""
counters = client.keys('counter:*:total')
pipe = client.pipeline(transaction=False)
for counter in counters:
pipe.get(counter)
return {
counter.split(':')[1]: int(value)
for counter, value in zip(counters, pipe.execute())
}
