From 474d9998e83930cd687451596cfd4ebd101f7069 Mon Sep 17 00:00:00 2001 From: Rick Zamora Date: Wed, 24 Jan 2024 09:58:13 -0600 Subject: [PATCH 1/2] set check_exact=False in failing test --- dask_expr/tests/test_quantiles.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/dask_expr/tests/test_quantiles.py b/dask_expr/tests/test_quantiles.py index 527c52f45..c5402087a 100644 --- a/dask_expr/tests/test_quantiles.py +++ b/dask_expr/tests/test_quantiles.py @@ -1,3 +1,5 @@ +from numpy import allclose + from dask_expr import from_pandas from dask_expr.tests._util import _backend_library, assert_eq @@ -12,7 +14,9 @@ def test_repartition_quantiles(): expected = pd.Series( [1, 1, 3, 7, 9, 15], index=[0, 0.2, 0.4, 0.6, 0.8, 1], name="a" ) - assert_eq(result, expected) + # Set `check_exact=False` explicitly + # See: https://github.com/dask-contrib/dask-expr/issues/801 + assert_eq(result, expected, check_exact=False) result = df.a._repartition_quantiles(npartitions=4) expected = pd.Series([1, 2, 5, 8, 15], index=[0, 0.25, 0.5, 0.75, 1], name="a") From 286e30c7d7f1dd14115e0db8495f5c7fcdd9981b Mon Sep 17 00:00:00 2001 From: Rick Zamora Date: Wed, 24 Jan 2024 10:47:11 -0600 Subject: [PATCH 2/2] formatting --- dask_expr/tests/test_quantiles.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dask_expr/tests/test_quantiles.py b/dask_expr/tests/test_quantiles.py index c5402087a..03ae4230a 100644 --- a/dask_expr/tests/test_quantiles.py +++ b/dask_expr/tests/test_quantiles.py @@ -1,5 +1,3 @@ -from numpy import allclose - from dask_expr import from_pandas from dask_expr.tests._util import _backend_library, assert_eq @@ -14,7 +12,7 @@ def test_repartition_quantiles(): expected = pd.Series( [1, 1, 3, 7, 9, 15], index=[0, 0.2, 0.4, 0.6, 0.8, 1], name="a" ) - # Set `check_exact=False` explicitly + # Set `check_exact=False` # See: https://github.com/dask-contrib/dask-expr/issues/801 assert_eq(result, expected, check_exact=False)