Adds basic test for square method.