diff --git a/tests/test_predict.py b/tests/test_predict.py --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -1,6 +1,6 @@ from unittest import TestCase -from languedoc.predict import preprocess, rank_ngram_freqs, Sample, identify +from languedoc.predict import preprocess, rank_ngram_counts, Sample, identify class TestPredict(TestCase): @@ -10,24 +10,24 @@ class TestPredict(TestCase): self.assertEqual(preprocess("1% "), " ") self.assertEqual(preprocess("Глава ĚŠČŘŽ"), " глава ěščřž ") - def test_rank_ngram_freqs(self): + def test_rank_ngram_counts(self): freqs = {"a": 3, "aa": 1, "b": 4, "bb": 1, "c": 1} expected = {"b": 0, "a": 1, "c": 2, "aa": 3, "bb": 4} - self.assertEqual(rank_ngram_freqs(freqs), expected) + self.assertEqual(rank_ngram_counts(freqs), expected) class TestSample(TestCase): def test_extract(self): a = Sample.extract("aaaaaa", "a") self.assertEqual(a.language, "a") - self.assertEqual(a.ranked_ngrams, {'a': 0, 'aa': 1, 'aaa': 2, ' aa': 3, 'aa ': 4, ' a': 5, 'a ': 6}) + self.assertEqual(a.ranked_ngrams, {'a': 0, 'aa': 1, 'aaa': 2, ' a': 3, 'a ': 4, ' aa': 5, 'aa ': 6}) b = Sample.extract("aa aa aa", "b") - self.assertEqual(b.ranked_ngrams, {'a': 0, ' aa': 1, 'aa ': 2, ' a': 3, 'a ': 4, 'aa': 5, 'a a': 6}) + self.assertEqual(b.ranked_ngrams, {'a': 0, ' a': 1, 'a ': 2, 'aa': 3, ' aa': 4, 'aa ': 5, 'a a': 6}) c = Sample.extract("aa") self.assertEqual(c.language, "??") - self.assertEqual(c.ranked_ngrams, {'a': 0, ' aa': 1, 'aa ': 2, ' a': 3, 'a ': 4, 'aa': 5}) + self.assertEqual(c.ranked_ngrams, {'a': 0, ' a': 1, 'a ': 2, 'aa': 3, ' aa': 4, 'aa ': 5}) class TestIdentify(TestCase):