ENH: a better repr for unyt arrays

yt-project · Sep 25, 2021 · 05bb1c7 · 05bb1c7
1 parent 51b3ddb
commit 05bb1c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/unyt/_array_functions.py b/unyt/_array_functions.py
@@ -16,7 +16,7 @@ def decorator(func):
 
 @implements(np.array2string)
 def array2string(a, *args, **kwargs):
-    return np.array2string._implementation(a, *args, **kwargs) + f" {a.units}"
+    return np.array2string._implementation(a, *args, **kwargs) + f", units={str(a.units)!r}"
 
 
 def _get_conversion_factor(out, units) -> float:

diff --git a/unyt/tests/test_array_functions.py b/unyt/tests/test_array_functions.py
@@ -10,7 +10,7 @@
 
 def test_array_repr():
     arr = [1, 2, 3] * cm
-    assert np.array_repr(arr) == "unyt_array([1, 2, 3] cm)"
+    assert np.array_repr(arr) == "unyt_array([1, 2, 3], units='cm')"
 
 
 def test_dot_vectors():