diff --git a/orchard_note.py b/orchard_note.py index 713dc42..1900ff2 100644 --- a/orchard_note.py +++ b/orchard_note.py @@ -8,7 +8,7 @@ from orchard_utils import to_base, to_scalar from utils import leos2bsp class OrchardNote(object): - def __init__(self, d, pk_d, v: Scalar, rho, rseed): + def __init__(self, d, pk_d, v, rho, rseed): self.d = d self.pk_d = pk_d self.v = v @@ -17,25 +17,27 @@ class OrchardNote(object): self.rcm = self.rcm(rho) self.psi = self.psi(rho) - def __bytes__(self): + def __eq__(self, other): + if other is None: + return False return ( - self.d + - bytes(self.pk_d) + - struct.pack('