fixed a small mistake in linalg's elementwise square unit test

@lambday, @lisitsyn Someone merge this please. Sorry about this!