Tweak import order

`py` is a third-party library so it belongs with `pytest` if we're separating stdlib from 3rd party.

I also alphabetized the stdlib imports cuz why not? :smile: