<gh_stars>1-10
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
#       OS : GNU/Linux Ubuntu 16.04 or 18.04
# LANGUAGE : Python 3.5.2 or later
#   AUTHOR : <NAME>.
#     DATE : 28.10.2019
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

'''
Tests for uk_stemmer.
'''

import time
import re
from uk_stemmer import UkStemmer


def main():
    stemmer = UkStemmer()
    test_string = 'Привіт,як твої  справи?  (це ж тест??) Зберігайте спокойствіе.Оплатіте ласка, зробіть погашення боргу/заборгованостей. ' + \
                  'Рефлексивного и тямущий'

    start_time = time.time()
    prepare_test_string = test_string.lower()
    words = re.split(r'(\W)', prepare_test_string)
    words = [word for word in words if word != '']

    for i in range(len(words)):
        words[i] = stemmer.stem_word(words[i])

    stem_test_string = ''.join(words)
    work_time = time.time() - start_time

    print('Source: %s\nStemmed: %s\nTime: %.6f' % (test_string, stem_test_string, work_time))

    etalon_stem_test_string = 'привіт,як тво  справ?  (це ж тест??) зберігайт спокойств.оплатіт ласк, зробіт погашенн борг/заборгованост. ' + \
                              'рефлексивн и тямущ'
    if stem_test_string == etalon_stem_test_string:
        print('ALL OK')


if __name__ == '__main__':
    main()
