Getting unit tests to work again
Migrating to pytest