def test_prune_sparse():
source = csc_matrix(test_matrices[4][0])
target = np.matrix(test_matrices[4][1])
pruned = mc.prune(source, 1).todense()
assert np.array_equal(pruned, target)
def test_prune():
source = np.matrix(test_matrices[4][0])
target = np.matrix(test_matrices[4][1])
pruned = mc.prune(source, 1)
assert np.array_equal(pruned, target)