()
| 2534 | |
| 2535 | |
| 2536 | def test_query(): |
| 2537 | df = pd.DataFrame({"x": [1, 2, 3, 4], "y": [5, 6, 7, 8]}) |
| 2538 | ddf = dd.from_pandas(df, npartitions=2) |
| 2539 | assert_eq(ddf.query("x**2 > y"), df.query("x**2 > y")) |
| 2540 | assert_eq( |
| 2541 | ddf.query("x**2 > @value", local_dict={"value": 4}), |
| 2542 | df.query("x**2 > @value", local_dict={"value": 4}), |
| 2543 | ) |
| 2544 | |
| 2545 | |
| 2546 | @pytest.mark.parametrize( |