from nw_insertion import NW_INSERTION

nw_insertion = NW_INSERTION()
a = 'ABCDEFGHIJKLMNOP'
b = ['QV','TR','Z']
print(a)
seq = nw_insertion.get_best_nw_based_insertions(a,b, False, verbose=True)
print('Final best is: ', seq)

# a = 'IGRHRYHIGG'
# b = 'SRYIGRG---'
# print('score', a, b, get_nw_score(a, b))

a = 'IGRHRYHIGG'
b = 'SRYIGRG'
seq = nw_insertion.get_best_nw_based_insertion(a,b)
print('Final best is: ', seq)

a = 'IGRHRYHIGG'
b = ['SRYIGRG']
print(a)
seq, _, _, _, _ = nw_insertion.get_best_nw_based_insertions(a,b, True, verbose=True)
print('Final best is: ', seq)

a = 'ABCDEFGHIJKLMNOP'
b = ['QV','TR','Z']
print(a)
seq = nw_insertion.get_nw_based_insertions_greedy(a, b, return_best=True, verbose=False)
print('Final best is: ', seq)