diff --git a/tests/test_curve.py b/tests/test_curve.py index 3c8f68a..3694728 100644 --- a/tests/test_curve.py +++ b/tests/test_curve.py @@ -1,5 +1,4 @@ import pytest - from umbral.curve import Curve, SECP256R1, SECP256K1, SECP384R1 diff --git a/umbral/curve.py b/umbral/curve.py index 614fec6..cadabcd 100644 --- a/umbral/curve.py +++ b/umbral/curve.py @@ -33,6 +33,9 @@ class Curve: def supported_curves(self): return self.__AVAIL_CURVES + def __eq__(self, other): + return self.curve_nid == other.curve_nid + class SECP256R1(Curve): """