N-gram distance between two strings

ngram_distance.py

def get_n_grams(s, n):
    n_grams = []
    i = 0
    while i + n <= len(s):
        n_grams.append(s[i:i + n])
        i += 1
    
    return set(n_grams)

def n_gram_distance(s1, s2, n, case_sensitive=False):
    if not case_sensitive:
        s1 = s1.lower()
        s2 = s2.lower()

    n_grams_s1 = get_n_grams(s1, n)
    n_grams_s2 = get_n_grams(s2, n)
    distance = 2 * len(n_grams_s1.intersection(n_grams_s2)) / (len(n_grams_s1) + len(n_grams_s2))

    print(f'{2}-grams')
    print(n_grams_s1)
    print(n_grams_s2)
    print(distance)