()
| 3724 | |
| 3725 | |
| 3726 | def test_broadcast_chunks(): |
| 3727 | assert broadcast_chunks() == () |
| 3728 | |
| 3729 | assert broadcast_chunks(((2, 3),)) == ((2, 3),) |
| 3730 | |
| 3731 | assert broadcast_chunks(((5, 5),), ((5, 5),)) == ((5, 5),) |
| 3732 | |
| 3733 | a = ((10, 10, 10), (5, 5)) |
| 3734 | b = ((5, 5),) |
| 3735 | assert broadcast_chunks(a, b) == ((10, 10, 10), (5, 5)) |
| 3736 | assert broadcast_chunks(b, a) == ((10, 10, 10), (5, 5)) |
| 3737 | |
| 3738 | a = ((10, 10, 10), (5, 5)) |
| 3739 | b = ((1,), (5, 5)) |
| 3740 | assert broadcast_chunks(a, b) == ((10, 10, 10), (5, 5)) |
| 3741 | |
| 3742 | a = ((10, 10, 10), (5, 5)) |
| 3743 | b = ((3, 3), (5, 5)) |
| 3744 | with pytest.raises(ValueError): |
| 3745 | broadcast_chunks(a, b) |
| 3746 | |
| 3747 | a = ((1,), (5, 5)) |
| 3748 | b = ((1,), (5, 5)) |
| 3749 | assert broadcast_chunks(a, b) == a |
| 3750 | |
| 3751 | a = ((1,), (np.nan, np.nan, np.nan)) |
| 3752 | b = ((3, 3), (1,)) |
| 3753 | r = broadcast_chunks(a, b) |
| 3754 | assert r[0] == b[0] and np.allclose(r[1], a[1], equal_nan=True) |
| 3755 | |
| 3756 | a = ((3, 3), (1,)) |
| 3757 | b = ((1,), (np.nan, np.nan, np.nan)) |
| 3758 | r = broadcast_chunks(a, b) |
| 3759 | assert r[0] == a[0] and np.allclose(r[1], b[1], equal_nan=True) |
| 3760 | |
| 3761 | a = ((3, 3), (5, 5)) |
| 3762 | b = ((1,), (np.nan, np.nan, np.nan)) |
| 3763 | with pytest.raises(ValueError): |
| 3764 | broadcast_chunks(a, b) |
| 3765 | |
| 3766 | |
| 3767 | def test_chunks_error(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…