diff --git a/tests/test_predict.py b/tests/test_predict.py --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -1,7 +1,8 @@ +import unittest from unittest import TestCase from languedoc.predict import preprocess, extract_kgram_counts, extract_ngram_counts, rank_ngram_counts, \ - extract_ranked_ngrams, Sample, identify + extract_ranked_ngrams, Sample, identify, sample_text class TestPredict(TestCase): @@ -11,6 +12,10 @@ class TestPredict(TestCase): self.assertEqual(preprocess("1% "), " ") self.assertEqual(preprocess("Глава ĚŠČŘŽ"), " глава ěščřž ") + @unittest.skip + def test_sample_text(self): + pass + def test_extract_kgram_counts(self): text = "abbbabb" self.assertEqual(extract_kgram_counts(text, 1), {"a": 2, "b": 5})