# =============================================================================
# Fog Levenshtein Keys Unit Tests
# =============================================================================
from fog.key import (
    levenshtein_1d_keys,
    damerau_levenshtein_1d_keys,
    levenshtein_1d_blocks,
    damerau_levenshtein_1d_blocks
)

HELLO_KEYS = set([
    'hello',
    '!ello',
    'h!llo',
    'he!lo',
    'hel!o',
    'hell!',
    '!hello',
    'h!ello',
    'he!llo',
    'hell!o',
    'hello!'
])

HELLO_TRANSPOSITION_KEYS = set([
    'ehllo',
    'hlelo',
    'helol'
])

HELLO_WORDS = [
    'BONJOUR',
    'BINJOUR',
    'BONTOUR',
    'NONJOUR',
    'ONJOUR',
    'BONJOU',
    'BONJOURE',
    'BONTJOUR',
    'TBONJOUR'
]

HELLO_WORDS_TRANSPOSITIONS = [
    'BOJNOUR',
    'BONOJUR',
    'OBNJOUR',
    'BONJORU'
]

SECOND_TRANSPOSITIONS_TEST = [
    'BONJOU',
    'OBNJOU',
    'BNOJOU',
    'BOJNOU',
    'BONOJU',
    'BONJUO',
    'BONJOUR',
    'TBONJOU',
    'BONTJOU',
    'BOTNJOU',
    'MONJOU'
]


class TestLevenshtein(object):
    def test_keys(self):

        keys = set(levenshtein_1d_keys('hello', flag='!'))

        assert keys == HELLO_KEYS

        keys_with_transpositions = set(levenshtein_1d_keys('hello', transpositions=True, flag='!'))

        assert keys_with_transpositions == HELLO_KEYS | HELLO_TRANSPOSITION_KEYS

        keys_with_transpositions = set(damerau_levenshtein_1d_keys('hello', flag='!'))

        assert keys_with_transpositions == HELLO_KEYS | HELLO_TRANSPOSITION_KEYS

    def test_blocks(self):
        bonjour_blocks = levenshtein_1d_blocks('BONJOUR')
        bonjou_blocks = damerau_levenshtein_1d_blocks('BONJOU')

        for word in HELLO_WORDS:
            blocks = levenshtein_1d_blocks(word)

            assert any(b in bonjour_blocks for b in blocks)

        for word in HELLO_WORDS + HELLO_WORDS_TRANSPOSITIONS:
            blocks = levenshtein_1d_blocks(word, transpositions=True)

            assert any(b in bonjour_blocks for b in blocks)

        for word in SECOND_TRANSPOSITIONS_TEST:
            blocks = damerau_levenshtein_1d_blocks(word)

            assert any(b in bonjou_blocks for b in blocks)

        # Small keys
        assert any(b in levenshtein_1d_blocks('h') for b in levenshtein_1d_blocks('he'))
        assert any(b in levenshtein_1d_blocks('h') for b in levenshtein_1d_blocks('eh'))
        assert any(b in levenshtein_1d_blocks('h') for b in levenshtein_1d_blocks('t'))
        assert any(b in levenshtein_1d_blocks('he') for b in levenshtein_1d_blocks('lhe'))