def test_reversed(self):
t = Trie()
t.add('')
t.add('abc')
t.add('a')
t.add('c')
tr = t.reversed()
self.assertCountEqual(list(tr),['','cba','a','c'])
def test_forward(self):
tx = Trie()
tx.add('')
tx.add('a')
ty = Trie()
ty.add('b')
ty.add('c')
ty.add('bc')
ty.add('')
ep = EditProbability(Alphabet(tx), Alphabet(ty))
ep.probs['a', ''] = 0.5
ep.probs['', 'b'] = 0.4
ep.probs['', 'c'] = 0.3
ep.probs['a', 'b'] = 0.2
ep.probs['', 'bc'] = 0.1
ep.probs['a', 'bc'] = 0.0
ep.probs['a', 'c'] = 0.0
r = ep.forward('a', 'bc')
self.assertAlmostEqual(r[1, 2], 0.34)
def test_oov(self):
tx = Trie()
tx.add('')
tx.add('a')
ty = Trie()
ty.add('b')
ty.add('')
ep = EditProbability(Alphabet(tx), Alphabet(ty))
self.assertAlmostEqual(ep.score('a', 'bc'), 0)
def test_score(self):
tx = Trie()
tx.add('')
tx.add('a')
ty = Trie()
ty.add('b')
ty.add('c')
ty.add('bc')
ty.add('')
ep = EditProbability(Alphabet(tx), Alphabet(ty))
ep.probs['a', ''] = 0.5
ep.probs['', 'b'] = 0.4
ep.probs['', 'c'] = 0.3
ep.probs['a', 'b'] = 0.2
ep.probs['', 'bc'] = 0.1
ep.probs['a', 'bc'] = 0.0
ep.probs['a', 'c'] = 0.0
self.assertAlmostEqual(ep.score('a', 'bc'), 0.34)
def test_add_contains_nested(self):
t = Trie()
t.add('test')
t.add('test2')
self.assertTrue(t.contains('test'))
self.assertTrue(t.contains('test2'))
def test_len(self):
t = Trie()
t.add('')
t.add('a')
self.assertEqual(len(t),2)
def test_iter(self):
t = Trie()
t.add('')
t.add('a')
self.assertCountEqual(t,['','a'])
def test_entries(self):
t = Trie()
t.add('')
t.add('test')
t.add('test2')
self.assertCountEqual(list(t),['','test','test2'])
def test_add_contains_epsilon(self):
t = Trie()
t.add('')
self.assertTrue(t.contains(''))
def test_prefixes_epsilon(self):
t = Trie()
t.add('')
self.assertCountEqual(t.prefixesOf(''),[''])
def test_prefixes(self):
t = Trie()
t.add('test')
t.add('test2')
t.add('test23')
t.add('test4')
self.assertCountEqual(t.prefixesOf('test23'),['test','test2','test23'])
def test_prefix_count(self):
t = Trie()
t.add('test')
t.add('test2')
self.assertEqual(t.prefixCount('test'),2)
def test_nocontain_prefixes(self):
t = Trie()
t.add('one')
self.assertFalse(t.contains('on'))
def test_add_contains_multiple(self):
t = Trie()
t.add('one')
t.add('two')
self.assertTrue(t.contains('one'))
self.assertTrue(t.contains('two'))
def test_init_trie(self):
t = Trie()
t.add('ab')
a = Alphabet(t)
self.assertCountEqual(a,['ab'])