Test infrastructure

I think we should use `pytest` for consitency with cryptography.