Modernize unit test style

Our current unit tests still use an old deprecated API and we get deprecation warnings on all of our test runs. We should switch to newer pytest `assert` style tests. We already use pytest as the test runner in our `tox.ini`, so it should be pretty trivial to switch.

We just need to be careful that we are still testing everything appropriately.