diff options
Diffstat (limited to 'test/ec/test_model.py')
| -rw-r--r-- | test/ec/test_model.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/test/ec/test_model.py b/test/ec/test_model.py index d5c5afe..e6bc176 100644 --- a/test/ec/test_model.py +++ b/test/ec/test_model.py @@ -1,3 +1,5 @@ +import pickle + from pyecsca.ec.model import ( ShortWeierstrassModel, MontgomeryModel, @@ -11,3 +13,14 @@ def test_load(): assert len(MontgomeryModel().coordinates) > 0 assert len(EdwardsModel().coordinates) > 0 assert len(TwistedEdwardsModel().coordinates) > 0 + + +def test_pickle(): + sw = ShortWeierstrassModel() + m = MontgomeryModel() + e = EdwardsModel() + te = TwistedEdwardsModel() + assert sw == pickle.loads(pickle.dumps(sw)) + assert m == pickle.loads(pickle.dumps(MontgomeryModel())) + assert e == pickle.loads(pickle.dumps(EdwardsModel())) + assert te == pickle.loads(pickle.dumps(TwistedEdwardsModel())) |
