Modernize unit test style

[elihunter173]

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.