diff options
author | warren <warren.weckesser@gmail.com> | 2021-12-03 22:37:07 -0500 |
---|---|---|
committer | warren <warren.weckesser@gmail.com> | 2021-12-03 22:37:07 -0500 |
commit | c975a3724831739c2a536510d3998a7f3033d149 (patch) | |
tree | 5304f69120dc53b52ea512fbeca22fd551c6dd90 /numpy/array_api/tests/test_array_object.py | |
parent | 6223584a5ec1e33b98619931e14460d54369dc0d (diff) | |
parent | a81535a364ca2d5aa277977e53c4e2302cae8ea2 (diff) | |
download | numpy-c975a3724831739c2a536510d3998a7f3033d149.tar.gz |
Merge branch 'main' into doc-fix-mvn-example
Diffstat (limited to 'numpy/array_api/tests/test_array_object.py')
-rw-r--r-- | numpy/array_api/tests/test_array_object.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/numpy/array_api/tests/test_array_object.py b/numpy/array_api/tests/test_array_object.py index 12479d765..deab50693 100644 --- a/numpy/array_api/tests/test_array_object.py +++ b/numpy/array_api/tests/test_array_object.py @@ -4,6 +4,7 @@ from numpy.testing import assert_raises import numpy as np from .. import ones, asarray, result_type, all, equal +from .._array_object import Array from .._dtypes import ( _all_dtypes, _boolean_dtypes, @@ -301,3 +302,16 @@ def test_device_property(): assert all(equal(asarray(a, device='cpu'), a)) assert_raises(ValueError, lambda: asarray(a, device='gpu')) + +def test_array_properties(): + a = ones((1, 2, 3)) + b = ones((2, 3)) + assert_raises(ValueError, lambda: a.T) + + assert isinstance(b.T, Array) + assert b.T.shape == (3, 2) + + assert isinstance(a.mT, Array) + assert a.mT.shape == (1, 3, 2) + assert isinstance(b.mT, Array) + assert b.mT.shape == (3, 2) |