Make test_hbavss actually fail on exceptions

Fix G1.one()
initc3 · Mar 2, 2019 · c409c60 · c409c60
1 parent c993aa8
commit c409c60
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/honeybadgermpc/betterpairing.py b/honeybadgermpc/betterpairing.py
@@ -72,6 +72,9 @@ def __imul__(self, other):
         if type(other) is G1:
             self.pyg1.add_assign(other.pyg1)
             return self
+        raise TypeError(
+            'Invalid multiplication param. Expected G1. Got '
+            + str(type(other)))
 
     def __truediv__(self, other):
         if type(other) is G1:
@@ -87,6 +90,9 @@ def __idiv__(self, other):
         if type(other) is G1:
             self.pyg1.sub_assign(other.pyg1)
             return self
+        raise TypeError(
+            'Invalid division param. Expected G1. Got '
+            + str(type(other)))
 
     def __pow__(self, other):
         if type(other) is int:
@@ -115,7 +121,7 @@ def __ipow__(self, other):
                 return self
             if other < 0:
                 self.invert()
-                other = other * -1
+                other *= -1
             self.pyg1.mul_assign(ZR(other).val)
             return self
         elif type(other) is ZR:
@@ -172,7 +178,9 @@ def pair_with(self, other):
 
     @staticmethod
     def one():
-        return G1(PyG1().zero())
+        one = G1()
+        one.pyg1.zero()
+        return one
 
     @staticmethod
     def rand(seed=None):
@@ -277,7 +285,7 @@ def __ipow__(self, other):
                 return self
             if other < 0:
                 self.invert()
-                other = other * -1
+                other *= -1
             self.pyg2.mul_assign(ZR(other).val)
             return self
         elif type(other) is ZR:
@@ -329,7 +337,9 @@ def projective(self):
 
     @staticmethod
     def one():
-        return G2(PyG2().zero())
+        one = G2()
+        one.pyg2.zero()
+        return one
 
     @staticmethod
     def rand(seed=None):
@@ -377,7 +387,7 @@ def __mul__(self, other):
                 return out
             if other < 0:
                 out.pyfq12.negate()
-                other = -1 * other
+                other *= -1
             prodend = GT(other)
             out.pyfq12.mul_assign(prodend.pyfq12)
             return out

diff --git a/tests/test_hbavss.py b/tests/test_hbavss.py
@@ -30,4 +30,4 @@ async def test_hbavss(test_router):
     for r in recipients:
         threads.append(r.run())
     threads.append(dealer.run())
-    await asyncio.wait(threads)
+    await asyncio.gather(*threads)