1 files changed, 113 insertions, 8 deletions

diff --git a/test/src/gen/test_gp.c b/test/src/gen/test_gp.c
index 0d01a3a..3f7a777 100644
--- a/test/src/gen/test_gp.c
+++ b/test/src/gen/test_gp.c
@@ -3,13 +3,14 @@
  * Copyright (C) 2017 J08nY
  */
 #include <criterion/criterion.h>
+#include <misc/types.h>
 #include "gen/gp.h"
 #include "test/default.h"
 #include "util/bits.h"
 
 TestSuite(gp, .init = default_setup, .fini = default_teardown);
 
-Test(gp, test_gp_seed) {
+Test(gp, test_gp_gen_seed) {
 	curve_t curve = {0};
 	config_t cfg = {};
 	arg_t arg = {.args = "() -> { return(Vecsmall([1,0])); }", .nargs = 1};
@@ -21,7 +22,7 @@ Test(gp, test_gp_seed) {
 	cr_assert(bits_eq(curve.seed->seed, bits_from_bin("10")),);
 }
 
-Test(gp, test_gp_field) {
+Test(gp, test_gp_gen_field) {
 	curve_t curve = {0};
 	config_t cfg = {};
 	arg_t arg = {.args = "(seed) -> { return(19); }", .nargs = 1};
@@ -31,7 +32,7 @@ Test(gp, test_gp_field) {
 	cr_assert(gequal(curve.field, stoi(19)),);
 }
 
-Test(gp, test_gp_a) {
+Test(gp, test_gp_gen_a) {
 	curve_t curve = {.field = stoi(19)};
 	config_t cfg = {};
 	arg_t arg = {.args = "(seed, field) -> { return(Mod(3,field)); }", .nargs = 1};
@@ -41,7 +42,7 @@ Test(gp, test_gp_a) {
 	cr_assert(gequal(curve.a, mkintmodu(3, 19)),);
 }
 
-Test(gp, test_gp_b) {
+Test(gp, test_gp_gen_b) {
 	curve_t curve = {.field = stoi(19), .a = mkintmodu(3, 19)};
 	config_t cfg = {};
 	arg_t arg = {.args = "(seed, field, a) -> { return(a * 2); }", .nargs = 1};
@@ -51,7 +52,7 @@ Test(gp, test_gp_b) {
 	cr_assert(gequal(curve.b, mkintmodu(6, 19)),);
 }
 
-Test(gp, test_gp_curve) {
+Test(gp, test_gp_gen_curve) {
 	curve_t curve = {.field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19)};
 	config_t cfg = {};
 	arg_t arg = {.args = "(seed, field, a, b) -> { return(ellinit([a,b], field)); }", .nargs = 1};
@@ -61,7 +62,7 @@ Test(gp, test_gp_curve) {
 	cr_assert(gequal(curve.curve, ellinit(mkvec2(curve.a, curve.b), curve.field, 0)),);
 }
 
-Test(gp, test_gp_order) {
+Test(gp, test_gp_gen_order) {
 	curve_t curve = {.field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19),
 	.curve = ellinit(mkvec2(stoi(3), stoi(6)), stoi(19), 0)};
 	config_t cfg = {};
@@ -72,7 +73,7 @@ Test(gp, test_gp_order) {
 	cr_assert(gequal(ellsea(curve.curve, 0), curve.order),);
 }
 
-Test(gp, test_gp_generators) {
+Test(gp, test_gp_gen_generators) {
 	curve_t curve = {.field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19),
 		.curve = ellinit(mkvec2(stoi(3), stoi(6)), stoi(19), 0), .order = stoi(16)
 	};
@@ -90,7 +91,7 @@ Test(gp, test_gp_generators) {
 	}
 }
 
-Test(gp, test_gp_points) {
+Test(gp, test_gp_gen_points) {
 	curve_t curve = {.field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19),
 		.curve = ellinit(mkvec2(stoi(3), stoi(6)), stoi(19), 0), .order = stoi(16),
 	};
@@ -106,4 +107,108 @@ Test(gp, test_gp_points) {
 	cr_assert_eq(ret, 1,);
 	cr_assert_eq(curve.npoints, 1,);
 	cr_assert(gequal(curve.points[0]->point, ellmul(curve.curve, gen.point, stoi(2))),);
+}
+
+Test(gp, test_gp_check_seed) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed) -> { return(1);}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_SEED);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_field) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19)};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field) -> { if(field == 19, return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_FIELD);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_a) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19)};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a) -> { if(a == Mod(3,19), return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_A);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_b) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(5, 19)};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a, b) -> { if(b == Mod(5,19), return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_B);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_curve) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(5, 19),
+		.curve = ellinit(mkvec2(stoi(3), stoi(5)), stoi(19), 0)};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a, b, curve) -> { if(curve == ellinit([3, 5], 19), return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_CURVE);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_order) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(5, 19),
+		.curve = ellinit(mkvec2(stoi(3), stoi(5)), stoi(19), 0), .order = stoi(16)};
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a, b, curve, order) -> { if(order == 16, return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_ORDER);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_generators) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19),
+		.curve = ellinit(mkvec2(stoi(3), stoi(6)), stoi(19), 0), .order = stoi(16),
+	};
+	point_t gen = {.point = mkvec2(mkintmodu(4,19), mkintmodu(14,19))};
+	point_t *generators[1] = {&gen};
+	curve.generators = generators;
+	curve.ngens = 1;
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a, b, curve, order, gens) -> { if(gens == ellgenerators(curve), return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_GENERATORS);
+	cr_assert_eq(ret, 1,);
+}
+
+Test(gp, test_gp_check_points) {
+	seed_t seed = {.seed = bits_from_hex("ff")};
+	curve_t curve = {.seed = &seed, .field = stoi(19), .a = mkintmodu(3, 19), .b = mkintmodu(6, 19),
+		.curve = ellinit(mkvec2(stoi(3), stoi(6)), stoi(19), 0), .order = stoi(16),
+	};
+	point_t gen = {.point = mkvec2(mkintmodu(4,19), mkintmodu(14,19))};
+	point_t *generators[1] = {&gen};
+	curve.generators = generators;
+	curve.ngens = 1;
+	curve.points = generators;
+	curve.npoints = 1;
+
+	config_t cfg = {};
+	arg_t arg = {.args="(seed, field, a, b, curve, order, gens, points) -> { if(points == ellgenerators(curve), return(1));}"};
+
+	int ret = gp_check(&curve, &cfg, &arg, OFFSET_POINTS);
+	cr_assert_eq(ret, 1,);
 }
\ No newline at end of file