wordfreq/tests/test_korean.py

from wordfreq import tokenize, word_frequency
import pytest


def test_tokens():
    assert tokenize("감사합니다", "ko") == ["감사", "합니다"]


def test_combination():
    gamsa_freq = word_frequency("감사", "ko")
    habnida_freq = word_frequency("합니다", "ko")

    assert word_frequency("감사감사", "ko") == pytest.approx(gamsa_freq / 2, rel=0.01)
    assert 1.0 / word_frequency("감사합니다", "ko") == pytest.approx(
        1.0 / gamsa_freq + 1.0 / habnida_freq, rel=0.01
    )