Fixed some pep8 issues in test_models

Change-Id: I2072429456061fe2f543cfe9547a77d659b96e34