def test_hmm_transition_matrix(self): hmm = HMM({INITIAL_STATE: ['q1'], 'q1': (['q2', 'q1', FINAL_STATE], ['dag', 'kot']), 'q2': (['q1', FINAL_STATE], ['z'])}) log_hmm(hmm) transition_matrix = hmm.get_transition_matrix() print(transition_matrix)