diff options
author | Jules Chéron <43635101+jules-ch@users.noreply.github.com> | 2023-03-22 21:31:15 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-22 21:31:15 +0100 |
commit | 3c6f5ffcb9304089caf4be07ded2c3644c2c0e13 (patch) | |
tree | a4b286d6053aa9e83ac1f81666886dac7a7c58dd /pint/testsuite | |
parent | d0106d78df7e39bfbc0ad1cd89f9e28d48e8af26 (diff) | |
parent | 90008e8402bde93b9c3fbb7d884d39f98ee6415c (diff) | |
download | pint-3c6f5ffcb9304089caf4be07ded2c3644c2c0e13.tar.gz |
Merge pull request #1722 from dopplershift/fix-dask
Fix __dask_postcompute__() to better preserve type
Diffstat (limited to 'pint/testsuite')
-rw-r--r-- | pint/testsuite/test_dask.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pint/testsuite/test_dask.py b/pint/testsuite/test_dask.py index 69c80fe..f4dee6a 100644 --- a/pint/testsuite/test_dask.py +++ b/pint/testsuite/test_dask.py @@ -149,6 +149,8 @@ def test_compute_persist_equivalent(local_registry, dask_array, numpy_array): assert np.all(res_compute == res_persist) assert res_compute.units == res_persist.units == units_ + assert type(res_compute) == local_registry.Quantity + assert type(res_persist) == local_registry.Quantity @pytest.mark.parametrize("method", ["compute", "persist", "visualize"]) |