def test_linear_index_load():
sig2 = utils.get_test_data('2.fa.sig')
sig47 = utils.get_test_data('47.fa.sig')
sig63 = utils.get_test_data('63.fa.sig')
ss2 = sourmash.load_one_signature(sig2, ksize=31)
ss47 = sourmash.load_one_signature(sig47)
ss63 = sourmash.load_one_signature(sig63)
with utils.TempDirectory() as location:
from sourmash import save_signatures
filename = os.path.join(location, 'foo')
with open(filename, 'wt') as fp:
sourmash.save_signatures([ss2, ss47, ss63], fp)
linear = LinearIndex.load(filename)
x = {ss2, ss47, ss63}
assert set(linear.signatures()) == x, linear.signatures
assert linear.filename == filename
def test_linear_index_save_load():
sig2 = utils.get_test_data('2.fa.sig')
sig47 = utils.get_test_data('47.fa.sig')
sig63 = utils.get_test_data('63.fa.sig')
ss2 = sourmash.load_one_signature(sig2, ksize=31)
ss47 = sourmash.load_one_signature(sig47)
ss63 = sourmash.load_one_signature(sig63)
linear = LinearIndex()
linear.insert(ss2)
linear.insert(ss47)
linear.insert(ss63)
with utils.TempDirectory() as location:
filename = os.path.join(location, 'foo')
linear.save(filename)
linear2 = LinearIndex.load(filename)
# now, search for sig2
sr = linear2.search(ss2, threshold=1.0)
print([s[1].name() for s in sr])
assert len(sr) == 1
assert sr[0][1] == ss2
