MCPcopy Create free account
hub / github.com/dask/dask / test_broadcast_to

Function test_broadcast_to

dask/array/tests/test_array_core.py:1344–1358  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1342
1343
1344def test_broadcast_to():
1345 x = np.random.default_rng().integers(10, size=(5, 1, 6))
1346 a = from_array(x, chunks=(3, 1, 3))
1347
1348 for shape in [a.shape, (5, 0, 6), (5, 4, 6), (2, 5, 1, 6), (3, 4, 5, 4, 6)]:
1349 xb = np.broadcast_to(x, shape)
1350 ab = broadcast_to(a, shape)
1351
1352 assert_eq(xb, ab)
1353
1354 if a.shape == ab.shape:
1355 assert a is ab
1356
1357 pytest.raises(ValueError, lambda: broadcast_to(a, (2, 1, 6)))
1358 pytest.raises(ValueError, lambda: broadcast_to(a, (3,)))
1359
1360
1361def test_broadcast_to_array():

Callers

nothing calls this directly

Calls 4

from_arrayFunction · 0.90
broadcast_toFunction · 0.90
assert_eqFunction · 0.90
integersMethod · 0.45

Tested by

no test coverage detected