add test for square_arrays_solution