Changeset - d76af898e537
[Not reviewed]
default
0 1 0
Laman - 2 years ago 2022-10-29 22:55:27

tests for exporting, loading and comparing
1 file changed with 17 insertions and 0 deletions:
0 comments (0 inline, 0 general)
tests/test_predict.py
Show inline comments
 
@@ -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):
0 comments (0 inline, 0 general)