(self)
| 5461 | |
| 5462 | @requires_sparse |
| 5463 | def test_from_dataframe_sparse(self) -> None: |
| 5464 | import sparse |
| 5465 | |
| 5466 | df_base = pd.DataFrame( |
| 5467 | {"x": range(10), "y": list("abcdefghij"), "z": np.arange(0, 100, 10)} |
| 5468 | ) |
| 5469 | |
| 5470 | ds_sparse = Dataset.from_dataframe(df_base.set_index("x"), sparse=True) |
| 5471 | ds_dense = Dataset.from_dataframe(df_base.set_index("x"), sparse=False) |
| 5472 | assert isinstance(ds_sparse["y"].data, sparse.COO) |
| 5473 | assert isinstance(ds_sparse["z"].data, sparse.COO) |
| 5474 | ds_sparse["y"].data = ds_sparse["y"].data.todense() |
| 5475 | ds_sparse["z"].data = ds_sparse["z"].data.todense() |
| 5476 | assert_identical(ds_dense, ds_sparse) |
| 5477 | |
| 5478 | ds_sparse = Dataset.from_dataframe(df_base.set_index(["x", "y"]), sparse=True) |
| 5479 | ds_dense = Dataset.from_dataframe(df_base.set_index(["x", "y"]), sparse=False) |
| 5480 | assert isinstance(ds_sparse["z"].data, sparse.COO) |
| 5481 | ds_sparse["z"].data = ds_sparse["z"].data.todense() |
| 5482 | assert_identical(ds_dense, ds_sparse) |
| 5483 | |
| 5484 | def test_to_and_from_empty_dataframe(self) -> None: |
| 5485 | # GH697 |
nothing calls this directly
no test coverage detected