def test_random_markov_chain_sparse():
sparse = True
n, k = 5, 3
mc_dicts = [{'P': random_markov_chain(n, sparse=sparse).P, 'k': n},
{'P': random_markov_chain(n, k, sparse=sparse).P, 'k': k}]
for mc_dict in mc_dicts:
P = mc_dict['P']
assert_array_equal(P.shape, (n, n))
assert_array_equal(P.getnnz(axis=1), [mc_dict['k']]*n)
def test_random_markov_chain_sparse():
sparse = True
n, k = 5, 3
mc_dicts = [{'P': random_markov_chain(n, sparse=sparse).P, 'k': n},
{'P': random_markov_chain(n, k, sparse=sparse).P, 'k': k}]
for mc_dict in mc_dicts:
P = mc_dict['P']
assert_array_equal(P.shape, (n, n))
assert_array_equal(P.getnnz(axis=1), [mc_dict['k']]*n)
