Update swift-dispersion-* checks

We should call the check with the .py extension as well as shebang
python in the file.