aboutsummaryrefslogtreecommitdiff
path: root/test/ec/test_model.py
blob: e6bc176a9b75c580795a484717d0d80060584dd6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import pickle

from pyecsca.ec.model import (
    ShortWeierstrassModel,
    MontgomeryModel,
    EdwardsModel,
    TwistedEdwardsModel,
)


def test_load():
    assert len(ShortWeierstrassModel().coordinates) > 0
    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()))