assert square of [1,2 and 3] equals 1,4,9