Skip to content

Commit

Permalink
Clarify comment on subclass
Browse files Browse the repository at this point in the history
Co-authored-by: Clément Robert <cr52@protonmail.com>
  • Loading branch information
kyleaoman and neutrinoceros authored Dec 19, 2024
1 parent 4bb98cd commit 9d46756
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion unyt/tests/test_array_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -1662,7 +1662,7 @@ def test_apply_over_axes(axes, expected_units):
# propagated correctly for well-behaved functions.
a = np.eye(3) * cm
ret = np.apply_over_axes(lambda x, axis: x[axis] ** 2, a, axes)
assert isinstance(ret, unyt_array) # could be subclass unyt_quantity
assert isinstance(ret, unyt_array) # could be unyt_quantity
assert ret.units == expected_units


Expand Down

0 comments on commit 9d46756

Please sign in to comment.