'Check that each stationary distribution sums to 1'

def test_sum_one(self):

    stationary_distributions = self.stationary
    assert_allclose(np.sum(stationary_distributions, axis=1), np.ones(self.n_stat_dists))
