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")