Multiple assertions for the square function