[square] adding tests.