| 19 | @pytest.mark.parametrize("sorted_index", [False, True]) |
| 20 | @pytest.mark.parametrize("sorted_map_index", [False, True]) |
| 21 | def test_series_map(base_npart, map_npart, sorted_index, sorted_map_index): |
| 22 | base = pd.Series(np.arange(100, dtype="i2")) |
| 23 | if not sorted_index: |
| 24 | index = np.arange(100) |
| 25 | np.random.shuffle(index) |
| 26 | base.index = index |
| 27 | map_index = np.arange(0, 1000, 10, dtype="i2") |
| 28 | mapper = pd.Series(np.random.randint(50, size=len(map_index)), index=map_index) |
| 29 | if not sorted_map_index: |
| 30 | map_index = np.array(map_index) |
| 31 | np.random.shuffle(map_index) |
| 32 | mapper.index = map_index |
| 33 | expected = base.map(mapper) |
| 34 | dask_base = dd.from_pandas(base, npartitions=base_npart, sort=False) |
| 35 | dask_map = dd.from_pandas(mapper, npartitions=map_npart, sort=False) |
| 36 | with pytest.warns(UserWarning, match="meta"): |
| 37 | result = dask_base.map(dask_map) |
| 38 | assert_eq(expected, result) |
| 39 | |
| 40 | |
| 41 | def test_series_map2(): |