Modify peer_id str

This commit is contained in:
Stuckinaboot 2019-05-21 22:25:08 -04:00
parent 4b40339b63
commit a4529d827d
2 changed files with 3 additions and 3 deletions

View File

@ -27,7 +27,7 @@ class ID:
def __str__(self):
pid = self.pretty()
return "<peer.ID %s>" % pid
return pid
__repr__ = __str__

View File

@ -36,7 +36,7 @@ def test_str_less_than_10():
for _ in range(5):
random_id_string += random.SystemRandom().choice(ALPHABETS)
pid = base58.b58encode(random_id_string).decode()
expected = "<peer.ID " + pid + ">"
expected = pid
actual = ID(random_id_string).__str__()
assert actual == expected
@ -46,7 +46,7 @@ def test_str_more_than_10():
for _ in range(10):
random_id_string += random.SystemRandom().choice(ALPHABETS)
pid = base58.b58encode(random_id_string).decode()
expected = "<peer.ID " + pid + ">"
expected = pid
actual = ID(random_id_string).__str__()
assert actual == expected