Convert unit tests to use Python's native unit tests