Migrate tests to use pytest

[alxmrs]

I don't like how tests are currently structured. I think a standard pytest flow would make them way better.

[alxmrs]

This is fixed in #76. Closing.