summaryrefslogtreecommitdiff
path: root/numpy/array_api/tests/test_array_object.py
diff options
context:
space:
mode:
authorwarren <warren.weckesser@gmail.com>2021-12-03 22:37:07 -0500
committerwarren <warren.weckesser@gmail.com>2021-12-03 22:37:07 -0500
commitc975a3724831739c2a536510d3998a7f3033d149 (patch)
tree5304f69120dc53b52ea512fbeca22fd551c6dd90 /numpy/array_api/tests/test_array_object.py
parent6223584a5ec1e33b98619931e14460d54369dc0d (diff)
parenta81535a364ca2d5aa277977e53c4e2302cae8ea2 (diff)
downloadnumpy-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.py14
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)