def test_get_taxid(): assert get_taxid('gi|455475748|gb|KC437649.1|') == P_venosa_id assert get_taxid('gi|455475748|gb|KC437649.1') == P_venosa_id assert get_taxid('KC437649') == P_venosa_id assert get_taxid('KC437649.1') == P_venosa_id assert get_taxid('gi|455475748') == P_venosa_id assert get_taxid('455475748') == P_venosa_id
#!/usr/local/bin/python import fileinput from BlastTools import get_taxid for line in fileinput.input(): if "Hit_ID" in line: continue fields = line.split() taxid = get_taxid(fields[2]) if taxid > 0: print '\t'.join(fields[0], taxid)