Added TODO to remove magic in Shock
pep8 cleanups in test_isentropic.py