()
| 285 | |
| 286 | |
| 287 | def test_broadcast_shape(): |
| 288 | # tests internal _broadcast_shape |
| 289 | # _broadcast_shape is already exercised indirectly by broadcast_arrays |
| 290 | # _broadcast_shape is also exercised by the public broadcast_shapes function |
| 291 | assert_equal(_broadcast_shape(), ()) |
| 292 | assert_equal(_broadcast_shape([1, 2]), (2,)) |
| 293 | assert_equal(_broadcast_shape(np.ones((1, 1))), (1, 1)) |
| 294 | assert_equal(_broadcast_shape(np.ones((1, 1)), np.ones((3, 4))), (3, 4)) |
| 295 | assert_equal(_broadcast_shape(*([np.ones((1, 2))] * 32)), (1, 2)) |
| 296 | assert_equal(_broadcast_shape(*([np.ones((1, 2))] * 100)), (1, 2)) |
| 297 | |
| 298 | # regression tests for gh-5862 |
| 299 | assert_equal(_broadcast_shape(*([np.ones(2)] * 32 + [1])), (2,)) |
| 300 | bad_args = [np.ones(2)] * 32 + [np.ones(3)] * 32 |
| 301 | assert_raises(ValueError, lambda: _broadcast_shape(*bad_args)) |
| 302 | |
| 303 | |
| 304 | def test_broadcast_shapes_succeeds(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…