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