Make KeyWallet hashable

Author: eunsoo-icon

iconsdk/wallet/wallet.py

@@ -191,6 +191,9 @@ def __ne__(self, other: KeyWallet) -> bool:
     def __deepcopy__(self, memodict={}) -> KeyWallet:
         return KeyWallet.load(self.private_key)
 
+    def __hash__(self):
+        return hash(self._private_key_object.secret)
+
 
 def public_key_to_address(public_key: bytes) -> str:
     if not (len(public_key) == 65 and public_key[0] == 4):

tests/wallet/test_wallet.py

@@ -1,5 +1,6 @@
 import copy
 from typing import Union, Dict
+from copy import deepcopy
 
 import pytest
 
@@ -70,6 +71,11 @@ def test_copy(self):
         wallet3 = copy.copy(wallet)
         assert wallet == wallet3
 
+    def test_hash(self):
+        wallet = KeyWallet.create()
+        wallet_dict = {wallet: wallet.public_key}
+        assert wallet.public_key == wallet_dict[wallet]
+
 
 def test_public_key_to_address():
     wallet: KeyWallet = KeyWallet.create()