def test_sparse_filter():
sps_filter = SparseFeatureFilter(threshold=0.5)
trans = sps_filter.fit_transform(sparse)
# we should have filtered out the third col
assert trans.shape[1] == 2
assert 'c' not in trans.columns
def test_sparse_filter_dense_data():
sps_filter = SparseFeatureFilter(threshold=0.25)
trans = sps_filter.fit_transform(iris)
# nothing should have changed
assert trans.equals(iris)
# assert on values
assert_array_almost_equal(sps_filter.sparsity_, np.zeros(4))
def test_sparse_asdf():
assert_transformer_asdf(SparseFeatureFilter(), iris)