Test get_wallet() function