Avoid scalar with dim > 0 by squeezing. Passes test and pep8 no warning.

Avoid scalar with dim > 0 by squeezing. Passes test and pep8 no warning. #999