def test_morpheme_boundary(self): self.configurations["MORPHEME_BOUNDARY_FLAG"] = True self.initialise_segment_table("plural_english_segment_table.txt") hmm = HMM({ INITIAL_STATE: ['q1'], 'q1': (['q2', FINAL_STATE], ['dog', 'kat']), 'q2': ([FINAL_STATE], ['z']) }) grammar = Grammar(hmm) self.assertCountEqual(['dog', 'kat', 'dogz', 'katz'], grammar.get_all_outputs())