| 123456789101112131415161718192021 |
- import numpy as np
- import pytest
- from pandas import SparseDataFrame, read_csv
- from pandas.util import testing as tm
- class TestSparseDataFrameToCsv(object):
- fill_values = [np.nan, 0, None, 1]
- @pytest.mark.parametrize('fill_value', fill_values)
- def test_to_csv_sparse_dataframe(self, fill_value):
- # GH19384
- sdf = SparseDataFrame({'a': type(self).fill_values},
- default_fill_value=fill_value)
- with tm.ensure_clean('sparse_df.csv') as path:
- sdf.to_csv(path, index=False)
- df = read_csv(path, skip_blank_lines=False)
- tm.assert_sp_frame_equal(df.to_sparse(fill_value=fill_value), sdf)
|
