Ejemplo n.º 1
0
 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)
Ejemplo n.º 5
0
 def test_add_contains_nested(self):
   t = Trie()
   t.add('test')
   t.add('test2')
   self.assertTrue(t.contains('test'))
   self.assertTrue(t.contains('test2'))
Ejemplo n.º 6
0
 def test_len(self):
   t = Trie()
   t.add('')
   t.add('a')
   self.assertEqual(len(t),2)
Ejemplo n.º 7
0
 def test_iter(self):
   t = Trie()
   t.add('')
   t.add('a')
   self.assertCountEqual(t,['','a'])
Ejemplo n.º 8
0
 def test_entries(self):
   t = Trie()
   t.add('')
   t.add('test')
   t.add('test2')
   self.assertCountEqual(list(t),['','test','test2'])
Ejemplo n.º 9
0
 def test_add_contains_epsilon(self):
   t = Trie()
   t.add('')
   self.assertTrue(t.contains(''))
Ejemplo n.º 10
0
 def test_prefixes_epsilon(self):
   t = Trie()
   t.add('')
   self.assertCountEqual(t.prefixesOf(''),[''])
Ejemplo n.º 11
0
 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'])
Ejemplo n.º 12
0
 def test_prefix_count(self):
   t = Trie()
   t.add('test')
   t.add('test2')
   self.assertEqual(t.prefixCount('test'),2)
Ejemplo n.º 13
0
 def test_nocontain_prefixes(self):
   t = Trie()
   t.add('one')
   self.assertFalse(t.contains('on'))
Ejemplo n.º 14
0
 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'])