def test_square(): y = energy.square(np.array([1, 2, 3])) assert_equal(y, 14) x = np.array([1, 2, 3, 4]) assert_equal(energy.square(x), 30)
def test_square(): y = energy.square(np.array([1, 2, 3])) assert_equal(y, 14)
def test_square(): x = np.array([1, 2, 3, 4]) assert_equal(energy.square(x), 30)