Fix unit test, per #36.
Change _linspace_square() method. #36