Corrected test_shapes according to PEP8