Merge pull request #1 from real-or-random/20210504_ge_set_gej_var

20210504 ge set gej var

src/tests.c

@@ -3103,8 +3103,13 @@ void test_ge(void) {
     for (i = 0; i < 4 * runs + 1; i++) {
         random_group_element_test(&ge[i]);
         /* randomly set half the points to infinity */
-        if(secp256k1_fe_is_odd(&ge[i].x)) {
-            secp256k1_ge_set_infinity(&ge[i]);
+        if (secp256k1_fe_is_odd(&ge[i].x) == i % 2) {
+          secp256k1_ge_set_infinity(&ge[i]);
+        }
+        int odd = secp256k1_fe_is_odd(&ge[i].x);
+        CHECK(odd == 0 || odd == 1);
+        if (!!secp256k1_fe_is_odd(&ge[i].x) == i % 2) {
+          secp256k1_ge_set_infinity(&ge[i]);
         }
         secp256k1_gej_set_ge(&gej[i], &ge[i]);
     }
@@ -3115,6 +3120,17 @@ void test_ge(void) {
         ge_equals_gej(&ge[i], &gej[i]);
     }
 
+    /* Test batch gej -> ge conversion with all infinities. */
+    for (i = 0; i < 4 * runs + 1; i++) {
+        secp256k1_gej_set_infinity(&gej[i]);
+    }
+    /* batch invert */
+    secp256k1_ge_set_all_gej_var(ge, gej, 4 * runs + 1);
+    /* check result */
+    for (i = 0; i < 4 * runs + 1; i++) {
+        ge_equals_gej(&ge[i], &gej[i]);
+    }
+
     free(ge);
     free(gej);
 }