()
| 277 | |
| 278 | |
| 279 | def test_broadcast_shape(): |
| 280 | # tests internal _broadcast_shape |
| 281 | # _broadcast_shape is already exercised indirectly by broadcast_arrays |
| 282 | # _broadcast_shape is also exercised by the public broadcast_shapes function |
| 283 | assert_equal(_broadcast_shape(), ()) |
| 284 | assert_equal(_broadcast_shape([1, 2]), (2,)) |
| 285 | assert_equal(_broadcast_shape(np.ones((1, 1))), (1, 1)) |
| 286 | assert_equal(_broadcast_shape(np.ones((1, 1)), np.ones((3, 4))), (3, 4)) |
| 287 | assert_equal(_broadcast_shape(*([np.ones((1, 2))] * 32)), (1, 2)) |
| 288 | assert_equal(_broadcast_shape(*([np.ones((1, 2))] * 100)), (1, 2)) |
| 289 | |
| 290 | # regression tests for gh-5862 |
| 291 | assert_equal(_broadcast_shape(*([np.ones(2)] * 32 + [1])), (2,)) |
| 292 | bad_args = [np.ones(2)] * 32 + [np.ones(3)] * 32 |
| 293 | assert_raises(ValueError, lambda: _broadcast_shape(*bad_args)) |
| 294 | |
| 295 | |
| 296 | def test_broadcast_shapes_succeeds(): |
nothing calls this directly
no test coverage detected