| 2513 | ], |
| 2514 | ) |
| 2515 | def test_polyfit_polyval_integration( |
| 2516 | use_dask: bool, x: xr.DataArray, y: xr.DataArray |
| 2517 | ) -> None: |
| 2518 | y.coords["x"] = x |
| 2519 | if use_dask: |
| 2520 | if not has_dask: |
| 2521 | pytest.skip("requires dask") |
| 2522 | y = y.chunk({"x": 2}) |
| 2523 | |
| 2524 | fit = y.polyfit(dim="x", deg=2) |
| 2525 | evaluated = xr.polyval(y.x, fit.polyfit_coefficients) |
| 2526 | expected = y.transpose(*evaluated.dims) |
| 2527 | xr.testing.assert_allclose(evaluated.variable, expected.variable) |
| 2528 | |
| 2529 | |
| 2530 | @pytest.mark.parametrize("use_dask", [False, True]) |