| 156 | |
| 157 | |
| 158 | static MatrixShape |
| 159 | _select_matrix_shape(PyArrayObject *array) |
| 160 | { |
| 161 | switch (PyArray_NDIM(array)) { |
| 162 | case 0: |
| 163 | return _scalar; |
| 164 | case 1: |
| 165 | if (PyArray_DIM(array, 0) > 1) |
| 166 | return _column; |
| 167 | return _scalar; |
| 168 | case 2: |
| 169 | if (PyArray_DIM(array, 0) > 1) { |
| 170 | if (PyArray_DIM(array, 1) == 1) |
| 171 | return _column; |
| 172 | else |
| 173 | return _matrix; |
| 174 | } |
| 175 | if (PyArray_DIM(array, 1) == 1) |
| 176 | return _scalar; |
| 177 | return _row; |
| 178 | } |
| 179 | return _matrix; |
| 180 | } |
| 181 | |
| 182 | |
| 183 | /* |
no test coverage detected