()
| 4828 | |
| 4829 | @pytest.mark.skip_with_pyarrow_strings # has to be array to explode |
| 4830 | def test_dataframe_explode(): |
| 4831 | df = pd.DataFrame({"A": [[1, 2, 3], "foo", [3, 4]], "B": 1}) |
| 4832 | exploded_df = df.explode("A") |
| 4833 | |
| 4834 | ddf = dd.from_pandas(df, npartitions=2) |
| 4835 | |
| 4836 | exploded_ddf = ddf.explode("A") |
| 4837 | assert ddf.divisions == exploded_ddf.divisions |
| 4838 | assert_eq(exploded_ddf.compute(), exploded_df) |
| 4839 | |
| 4840 | |
| 4841 | @pytest.mark.skip_with_pyarrow_strings # has to be array to explode |