def test_get_table(self):
# run a unit
# get a result
res = markov.get_table('xy')
# make assertion about result
self.assertEqual(res, {'x': {'y': 1}})
def test_table(self):
lines = ['abc', 'def', 'hi']
t = mar.get_table(char_gen(lines))
# print(t)
self.assertEqual(
t, {
'a': {
'b': 1
},
'b': {
'c': 1
},
'c': {
'd': 1
},
'd': {
'e': 1
},
'e': {
'f': 1
},
'f': {
'h': 1
},
'h': {
'i': 1
}
})
def test_get_table(self):
res = markov.get_table('abacab')
self.assertEqual(res, {
'a': {
'b': 2,
'c': 1
},
'b': {
'a': 1
},
'c': {
'a': 1
}
})
def test_get_table2(self):
res = markov.get_table('abacab', size=2)
self.assertEqual(res, {
'ab': {
'a': 1
},
'ba': {
'c': 1
},
'ac': {
'a': 1
},
'ca': {
'b': 1
}
})
def test_get_table(self):
lines = ['abc', 'def']
t = mar.get_table(mar.char_gen(lines))
self.assertEqual(
t, {
'a': {
'b': 1
},
'b': {
'c': 1
},
'c': {
'd': 1
},
'd': {
'e': 1
},
'e': {
'f': 1
}
})
def test_get_table2(self):
res = mk.get_table('abc', size=2)
self.assertEqual(res, {'ab': {'c': 1}})
def test_get_table(self):
res = mk.get_table('ab')
self.assertEqual(res, {'a': {'b': 1}})
def test_get_table_word(self):
#import pdb; pdb.set_trace()
res = markov.get_table(['hello', 'world'])
self.assertEqual(res, {'hello': {'world': 1}})
def test_get_table2(self):
res = markov.get_table('abcd', size=2)
self.assertEqual(res, {'ab': {'c': 1}, 'bc': {'d': 1}})
def test_get_table_word(self):
res = mk.get_table(['hello', 'world'], word=True)
self.assertEqual(res, {'hello': {'world': 1}})
