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

Function test_broadcast_to

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

Source from the content-addressed store, hash-verified

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