| 30 | |
| 31 | |
| 32 | def test_array_attributes(): |
| 33 | a = np.array(0, "f8") |
| 34 | assert m.ndim(a) == 0 |
| 35 | assert all(m.shape(a) == []) |
| 36 | assert all(m.strides(a) == []) |
| 37 | with pytest.raises(IndexError) as excinfo: |
| 38 | m.shape(a, 0) |
| 39 | assert str(excinfo.value) == "invalid axis: 0 (ndim = 0)" |
| 40 | with pytest.raises(IndexError) as excinfo: |
| 41 | m.strides(a, 0) |
| 42 | assert str(excinfo.value) == "invalid axis: 0 (ndim = 0)" |
| 43 | assert m.writeable(a) |
| 44 | assert m.size(a) == 1 |
| 45 | assert m.itemsize(a) == 8 |
| 46 | assert m.nbytes(a) == 8 |
| 47 | assert m.owndata(a) |
| 48 | |
| 49 | a = np.array([[1, 2, 3], [4, 5, 6]], "u2").view() |
| 50 | a.flags.writeable = False |
| 51 | assert m.ndim(a) == 2 |
| 52 | assert all(m.shape(a) == [2, 3]) |
| 53 | assert m.shape(a, 0) == 2 |
| 54 | assert m.shape(a, 1) == 3 |
| 55 | assert all(m.strides(a) == [6, 2]) |
| 56 | assert m.strides(a, 0) == 6 |
| 57 | assert m.strides(a, 1) == 2 |
| 58 | with pytest.raises(IndexError) as excinfo: |
| 59 | m.shape(a, 2) |
| 60 | assert str(excinfo.value) == "invalid axis: 2 (ndim = 2)" |
| 61 | with pytest.raises(IndexError) as excinfo: |
| 62 | m.strides(a, 2) |
| 63 | assert str(excinfo.value) == "invalid axis: 2 (ndim = 2)" |
| 64 | assert not m.writeable(a) |
| 65 | assert m.size(a) == 6 |
| 66 | assert m.itemsize(a) == 2 |
| 67 | assert m.nbytes(a) == 12 |
| 68 | assert not m.owndata(a) |
| 69 | |
| 70 | |
| 71 | @pytest.mark.skipif(not hasattr(m, "shape_span"), reason="std::span not available") |