add test for #78

fplll · Jul 12, 2017 · 4bcaadd · 4bcaadd
1 parent 557871e
commit 4bcaadd
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_strategies.py b/tests/test_strategies.py
@@ -0,0 +1,15 @@
+from fpylll import IntegerMatrix, BKZ
+from fpylll.fplll.bkz_param import Pruning, Strategy
+
+
+def test_linear_pruning():
+    A = IntegerMatrix.random(25, "qary", k=15, q=127)
+    block_size  = 10
+    preprocessing = 3
+    strategies = [Strategy(i) for i in range(5)]
+
+    for b in range(5, block_size+1):
+        strategies.append(Strategy(b, [preprocessing], [Pruning.LinearPruning(b, 2)]))
+
+    param = BKZ.Param(block_size=block_size, strategies=strategies)
+    BKZ.reduction(A, param)