Skip to content

Commit 4745546

Browse files
committed
Fix broken test cases
1 parent 27f8426 commit 4745546

2 files changed

Lines changed: 6 additions & 21 deletions

File tree

tests/misc/test_cprf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def test_cprf_1(self):
1212
cprf.setup_key()
1313
keys = [cprf.keygen(i) for i in range(n)]
1414
x = b'hello'
15-
res = [cprf.eval(i, keys[i], x, 512) for i in range(n)]
15+
res = [cprf.eval(n, i, keys[i], x, 512) for i in range(n)]
1616
assert sum(res) == 0
1717

1818
def test_cprf_2(self):
@@ -21,5 +21,5 @@ def test_cprf_2(self):
2121
cprf.setup_key()
2222
keys = [cprf.keygen(i) for i in range(n)]
2323
x = b'123456'
24-
res = [cprf.eval(i, keys[i], x, 512) for i in range(n)]
24+
res = [cprf.eval(n, i, keys[i], x, 512) for i in range(n)]
2525
assert sum(res) == 0

tests/multiclient/test_damgard.py

Lines changed: 4 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -8,21 +8,6 @@
88
class TestFeDamgardMultiClient(TestBase):
99
logging.getLogger().setLevel(logging.INFO)
1010

11-
def test_export(self):
12-
n = 3
13-
m = 5
14-
x = [[i + j for j in range(m)] for i in range(n)]
15-
y = [[i - j + 10 for j in range(m)] for i in range(n)]
16-
tag = b"testingtag123"
17-
key = FeDamgardMultiClient.generate(n, m)
18-
cs = [FeDamgardMultiClient.encrypt(x[i], tag, key.get_enc_key(i)) for i in range(n)]
19-
sk = FeDamgardMultiClient.keygen(y, key)
20-
json.dumps([key.get_enc_key(i).export() for i in range(n)])
21-
json.dumps(key.export())
22-
[json.dumps(cs[i].export()) for i in range(n)]
23-
json.dumps(sk.export())
24-
json.dumps(key.get_public_key().export())
25-
2611
def test_scheme_1(self):
2712
start = time.time()
2813
n = 3
@@ -45,8 +30,8 @@ def test_scheme_1(self):
4530

4631
def test_scheme_2(self):
4732
start = time.time()
48-
n = 25
49-
m = 25
33+
n = 10
34+
m = 10
5035
x = [[i * 10 + j for j in range(m)] for i in range(n)]
5136
y = [[i - j - 5 for j in range(m)] for i in range(n)]
5237
tag = str(start).encode()
@@ -66,8 +51,8 @@ def test_scheme_2(self):
6651

6752
def test_scheme_3(self):
6853
start = time.time()
69-
n = 50
70-
m = 50
54+
n = 25
55+
m = 25
7156
x = [[1 for j in range(m)] for i in range(n)]
7257
y = [[1 for j in range(m)] for i in range(n)]
7358
tag = str(start).encode()

0 commit comments

Comments
 (0)