def test_expand_sparse(): source = csc_matrix(test_matrices[2][0]) target = np.matrix(test_matrices[2][1]) expanded = mc.expand(source, 2).todense() assert np.array_equal(expanded, target)
def test_expand(): source = np.matrix(test_matrices[2][0]) target = np.matrix(test_matrices[2][1]) expanded = mc.expand(source, 2) assert np.array_equal(expanded, target)