Make KeyWallet hashable

icon-project · Apr 27, 2023 · ee69c20 · ee69c20
1 parent a4da7e3
commit ee69c20
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/iconsdk/wallet/wallet.py b/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):

diff --git a/tests/wallet/test_wallet.py b/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()