(input_subset, expected_dict, expected_index)
| 264 | ], |
| 265 | ) |
| 266 | def test_multi_columns(input_subset, expected_dict, expected_index): |
| 267 | # GH 39240 |
| 268 | df = pd.DataFrame( |
| 269 | { |
| 270 | "A": [[0, 1, 2], np.nan, [], (3, 4), np.nan], |
| 271 | "B": 1, |
| 272 | "C": [["a", "b", "c"], "foo", [], ["d", "e"], np.nan], |
| 273 | }, |
| 274 | index=list("abcde"), |
| 275 | ) |
| 276 | result = df.explode(input_subset) |
| 277 | expected = pd.DataFrame(expected_dict, expected_index) |
| 278 | tm.assert_frame_equal(result, expected) |
| 279 | |
| 280 | |
| 281 | def test_multi_columns_nan_empty(): |