def test_should_return_next_state_when_transition_probability_is_one(self):   
		markov_chain = MarkovChain([1,2], [[0,1],
										   [1,0]])
		next_state = markov_chain.forecast_next(1)
		self.assertEquals(2, next_state)
	def test_should_stay_in_current_state_when_transition_probability_is_zero(self):   
		markov_chain = MarkovChain([1,2], [[1,0],
										   [1,0]])
		next_state = markov_chain.forecast_next(1)
		self.assertEquals(1, next_state)