def test_names(): assert Add(x, 1)._name == x._name assert Add(1, x)._name == x._name assert Mult(Add(1, x), 2)._name == x._name assert Add(y, x)._name != x._name assert Add(y, x)._name != y._name assert Add(x, x)._name == x._name
def test_arithmetic_broadcasts_to_scalars(): assert Add(x, a).shape == x.shape assert Add(x, 1).shape == x.shape
def test_arithmetic_dshape_on_collections(): assert Add(x, y).shape == x.shape == y.shape