class TestTrain(unittest.TestCase):
def setUp(self):
self.mark = Markov()
self.model = self.mark.train_model(
open(os.getcwd() + "\\samples\\" + "TEST.txt", "r").read())
self.model_higher = self.mark.train_model_higher(
open(os.getcwd() + "\\samples\\" + "TEST.txt", "r").read())
def test_len(self):
self.assertTrue(len(self.model) != 0, "Len must be >=0 ")
def test_model(self):
self.assertEquals(len(self.model), len(standard_first_level))
def test_model_higher(self):
self.assertEquals(len(self.model_higher), len(standard_second_level))
def test_random_begin(self):
self.assertTrue(
self.mark.generate_start(self.model) in self.model['.'],
"Error in generate_start")
