()
| 4789 | |
| 4790 | @pytest.mark.skip_with_pyarrow_strings # has to be array to explode |
| 4791 | def test_dataframe_explode(): |
| 4792 | df = pd.DataFrame({"A": [[1, 2, 3], "foo", [3, 4]], "B": 1}) |
| 4793 | exploded_df = df.explode("A") |
| 4794 | |
| 4795 | ddf = dd.from_pandas(df, npartitions=2) |
| 4796 | |
| 4797 | exploded_ddf = ddf.explode("A") |
| 4798 | assert ddf.divisions == exploded_ddf.divisions |
| 4799 | assert_eq(exploded_ddf.compute(), exploded_df) |
| 4800 | |
| 4801 | |
| 4802 | @pytest.mark.skip_with_pyarrow_strings # has to be array to explode |