|
7 | 7 |
|
8 | 8 |
|
9 | 9 | @pytest.mark.parametrize(
|
10 |
| - "vals_left, vals_right", |
| 10 | + "vals_left, vals_right, dtype", |
11 | 11 | [
|
12 |
| - ([1, 2, 3], [1, 2]), |
13 |
| - (["a", "b", "c"], ["a", "b"]), |
| 12 | + ([1, 2, 3], [1, 2], "int64"), |
| 13 | + (["a", "b", "c"], ["a", "b"], "object"), |
14 | 14 | pytest.param(
|
15 |
| - pd.Series(["a", "b", "c"], dtype="string[pyarrow]"), |
16 |
| - pd.Series(["a", "b"], dtype="string[pyarrow]"), |
| 15 | + ["a", "b", "c"], |
| 16 | + ["a", "b"], |
| 17 | + "string[pyarrow]", |
17 | 18 | marks=td.skip_if_no("pyarrow"),
|
18 | 19 | ),
|
19 | 20 | ],
|
20 | 21 | )
|
21 |
| -def test_leftsemi(vals_left, vals_right): |
| 22 | +def test_leftsemi(vals_left, vals_right, dtype): |
| 23 | + vals_left = pd.Series(vals_left, dtype=dtype) |
| 24 | + vals_right = pd.Series(vals_right, dtype=dtype) |
22 | 25 | left = pd.DataFrame({"a": vals_left, "b": [1, 2, 3]})
|
23 | 26 | right = pd.DataFrame({"a": vals_right, "c": 1})
|
24 | 27 | expected = pd.DataFrame({"a": vals_right, "b": [1, 2]})
|
|
0 commit comments