()
| 240 | |
| 241 | |
| 242 | def test_broadcast_to_succeeds(): |
| 243 | data = [ |
| 244 | [np.array(0), (0,), np.array(0)], |
| 245 | [np.array(0), (1,), np.zeros(1)], |
| 246 | [np.array(0), (3,), np.zeros(3)], |
| 247 | [np.ones(1), (1,), np.ones(1)], |
| 248 | [np.ones(1), (2,), np.ones(2)], |
| 249 | [np.ones(1), (1, 2, 3), np.ones((1, 2, 3))], |
| 250 | [np.arange(3), (3,), np.arange(3)], |
| 251 | [np.arange(3), (1, 3), np.arange(3).reshape(1, -1)], |
| 252 | [np.arange(3), (2, 3), np.array([[0, 1, 2], [0, 1, 2]])], |
| 253 | # test if shape is not a tuple |
| 254 | [np.ones(0), 0, np.ones(0)], |
| 255 | [np.ones(1), 1, np.ones(1)], |
| 256 | [np.ones(1), 2, np.ones(2)], |
| 257 | # these cases with size 0 are strange, but they reproduce the behavior |
| 258 | # of broadcasting with ufuncs (see test_same_as_ufunc above) |
| 259 | [np.ones(1), (0,), np.ones(0)], |
| 260 | [np.ones((1, 2)), (0, 2), np.ones((0, 2))], |
| 261 | [np.ones((2, 1)), (2, 0), np.ones((2, 0))], |
| 262 | ] |
| 263 | for input_array, shape, expected in data: |
| 264 | actual = broadcast_to(input_array, shape) |
| 265 | assert_array_equal(expected, actual) |
| 266 | |
| 267 | |
| 268 | def test_broadcast_to_raises(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…