diff --git a/tests/test_predict.py b/tests/test_predict.py --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -55,6 +55,23 @@ class TestSample(TestCase): self.assertEqual(c.language, "??") self.assertEqual(c.ranked_ngrams, {'a': 0, ' a': 1, 'a ': 2, 'aa': 3, ' aa': 4, 'aa ': 5}) + def test_load(self): + exported = {"language": "en", "ngrams": list("abcdefgh")} + a = Sample.load(exported) + self.assertEqual(a.language, "en") + self.assertEqual(a.ranked_ngrams, {k: 7-v for (k, v) in zip("hgfedcba", range(8))}) + + def test_export(self): + a = Sample("en", {k: 7-v for (k, v) in zip("hgfedcba", range(8))}) + self.assertEqual(a.export(), {"language": "en", "ngrams": list("abcdefgh")}) + + def test_compare(self): + a = Sample("ab", {"a": 0, "b": 1, "ab": 2}) + b = Sample("bc", {"b": 0, "ba": 1, "a": 2, "ab": 3}) + + self.assertEqual(a.compare(b), 4) + self.assertEqual(b.compare(a), 7) + class TestIdentify(TestCase): def test_identify(self):