Remove str.removesuffix

1 files changed, 6 insertions, 4 deletions

diff --git a/pyecsca/ec/formula/switch_sign.py b/pyecsca/ec/formula/switch_sign.py
index bdf31e2..1acef5b 100644
--- a/pyecsca/ec/formula/switch_sign.py
+++ b/pyecsca/ec/formula/switch_sign.py
@@ -51,13 +51,15 @@ def sign_test(output_signs: Dict[str, int], coordinate_model: Any):
     p = 7
     out_inds = set(map(lambda x: "".join([o for o in x if o.isdigit()]), output_signs))
     for ind in out_inds:
-        point = {}
+        point_dict = {}
         for out, sign in output_signs.items():
-            out_var = out.removesuffix(ind)
+            if not out.endswith(ind):
+                continue
+            out_var = out[:out.index(ind)]
             if not out_var.isalpha():
                 continue
-            point[out_var] = Mod(sign, p)
-        point = Point(coordinate_model, **point)
+            point_dict[out_var] = Mod(sign, p)
+        point = Point(coordinate_model, **point_dict)
         try:
             apoint = point.to_affine()
         except NotImplementedError: