Fix lint.

1 files changed, 40 insertions, 19 deletions

diff --git a/test/ec/test_countermeasures.py b/test/ec/test_countermeasures.py
index 35ac5e8..3f78f74 100644
--- a/test/ec/test_countermeasures.py
+++ b/test/ec/test_countermeasures.py
@@ -1,8 +1,12 @@
 import pytest
 
 from pyecsca.ec.mult import LTRMultiplier
-from pyecsca.ec.countermeasures import GroupScalarRandomization, AdditiveSplitting, MultiplicativeSplitting, \
-    EuclideanSplitting
+from pyecsca.ec.countermeasures import (
+    GroupScalarRandomization,
+    AdditiveSplitting,
+    MultiplicativeSplitting,
+    EuclideanSplitting,
+)
 
 
 @pytest.fixture(params=["add-1998-cmo-2", "add-2015-rcb"])
@@ -14,15 +18,20 @@ def add(secp128r1, request):
 def dbl(secp128r1, request):
     return secp128r1.curve.coordinate_model.formulas[request.param]
 
+
 @pytest.fixture()
 def mult(secp128r1, add, dbl):
     return LTRMultiplier(add, dbl, complete=False)
 
+
 @pytest.mark.parametrize(
-    "num", [325385790209017329644351321912443757746,
-            123456789314159265358979323846264338327,
-            987654321314159265358979323846264338327,
-            786877845665557891354654531354008066400]
+    "num",
+    [
+        325385790209017329644351321912443757746,
+        123456789314159265358979323846264338327,
+        987654321314159265358979323846264338327,
+        786877845665557891354654531354008066400,
+    ],
 )
 def test_group_scalar_rand(mult, secp128r1, num):
     mult.init(secp128r1, secp128r1.generator)
@@ -33,11 +42,15 @@ def test_group_scalar_rand(mult, secp128r1, num):
     masked = gsr.multiply(num)
     assert raw.equals(masked)
 
+
 @pytest.mark.parametrize(
-    "num", [325385790209017329644351321912443757746,
-            123456789314159265358979323846264338327,
-            987654321314159265358979323846264338327,
-            786877845665557891354654531354008066400]
+    "num",
+    [
+        325385790209017329644351321912443757746,
+        123456789314159265358979323846264338327,
+        987654321314159265358979323846264338327,
+        786877845665557891354654531354008066400,
+    ],
 )
 def test_additive_splitting(mult, secp128r1, num):
     mult.init(secp128r1, secp128r1.generator)
@@ -48,11 +61,15 @@ def test_additive_splitting(mult, secp128r1, num):
     masked = asplit.multiply(num)
     assert raw.equals(masked)
 
+
 @pytest.mark.parametrize(
-    "num", [325385790209017329644351321912443757746,
-            123456789314159265358979323846264338327,
-            987654321314159265358979323846264338327,
-            786877845665557891354654531354008066400]
+    "num",
+    [
+        325385790209017329644351321912443757746,
+        123456789314159265358979323846264338327,
+        987654321314159265358979323846264338327,
+        786877845665557891354654531354008066400,
+    ],
 )
 def test_multiplicative_splitting(mult, secp128r1, num):
     mult.init(secp128r1, secp128r1.generator)
@@ -63,11 +80,15 @@ def test_multiplicative_splitting(mult, secp128r1, num):
     masked = msplit.multiply(num)
     assert raw.equals(masked)
 
+
 @pytest.mark.parametrize(
-    "num", [325385790209017329644351321912443757746,
-            123456789314159265358979323846264338327,
-            987654321314159265358979323846264338327,
-            786877845665557891354654531354008066400]
+    "num",
+    [
+        325385790209017329644351321912443757746,
+        123456789314159265358979323846264338327,
+        987654321314159265358979323846264338327,
+        786877845665557891354654531354008066400,
+    ],
 )
 def test_euclidean_splitting(mult, secp128r1, num):
     mult.init(secp128r1, secp128r1.generator)
@@ -76,4 +97,4 @@ def test_euclidean_splitting(mult, secp128r1, num):
     esplit = EuclideanSplitting(mult)
     esplit.init(secp128r1, secp128r1.generator)
     masked = esplit.multiply(num)
-    assert raw.equals(masked)
\ No newline at end of file
+    assert raw.equals(masked)