def test_shape_broadcast():
c = minitorch.shape_broadcast((1,), (5, 5))
assert c == (5, 5)
c = minitorch.shape_broadcast((5, 5), (1,))
assert c == (5, 5)
c = minitorch.shape_broadcast((1, 5, 5), (5, 5))
assert c == (1, 5, 5)
c = minitorch.shape_broadcast((5, 1, 5, 1), (1, 5, 1, 5))
assert c == (5, 5, 5, 5)
with pytest.raises(minitorch.IndexingError):
c = minitorch.shape_broadcast((5, 7, 5, 1), (1, 5, 1, 5))
print(c)
with pytest.raises(minitorch.IndexingError):
c = minitorch.shape_broadcast((5, 2), (5,))
print(c)
c = minitorch.shape_broadcast((2, 5), (5,))
assert c == (2, 5)
import minitorch c = minitorch.shape_broadcast((1, ), (1, )) print(c)