DOC: adjust doctests to changes in array repr from numpy 2.0

docs/usage.rst

@@ -1494,7 +1494,7 @@ Methods that hang off of a :mod:`unyt_dask_array` object and operations on
 One important caveat is that using Dask array functions may strip units:
 
     >>> da.sum(x_da).compute()
-    49995000
+    np.int64(49995000)
 
 For simple reductions, you can use the :mod:`reduce_with_units` function:
 

tox.ini

@@ -20,8 +20,8 @@ depends = begin
 deps =
     pytest
     h5py
-    pint
-    astropy
+    !py39: pint
+    !py39: astropy!=6.1.1
     coverage[toml]>=5.0
     pytest-cov
     pytest-doctestplus
@@ -32,6 +32,14 @@ commands =
     pytest --cov=unyt --cov-append --doctest-modules --doctest-plus --doctest-rst --basetemp={envtmpdir}
     coverage report --omit='.tox/*'
 
+[testenv:py39]
+# skip doctest on py39 because doctests require numpy>=2.0 and all optional deps,
+# but some of our optional deps (pint, astropy) don't have a version that support
+# both numpy>=2.0 and Python 3.9
+commands=
+    pytest --cov=unyt --cov-append --basetemp={envtmpdir}
+    coverage report --omit='.tox/*'
+
 [testenv:py39-versions]
 deps =
     docutils

unyt/array.py

@@ -1255,7 +1255,7 @@ def argsort(self, axis=-1, kind="quicksort", order=None):
         >>> from unyt import km
         >>> data = [3, 8, 7]*km
         >>> print(np.argsort(data))
-        [0 2 1]
+        [0 2 1] km
         >>> print(data.argsort())
         [0 2 1]
         """