Add test cases for Wallet class