Add tests to the wallet.py module.

This increases test coverage a lot.