Pytest

Ported the test suite to pytest