Update tests.

world-federation-of-advertisers · Jan 5, 2024 · 080dfec · 080dfec
1 parent 36f39ac
commit 080dfec
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/cc/any_sketch/crypto/secret_share_generator_test.cc b/src/test/cc/any_sketch/crypto/secret_share_generator_test.cc
@@ -67,6 +67,9 @@ TEST(AdditiveSecretSharing, SecretShareOverRingZ2kElementsSucceeds) {
       std::vector<uint32_t> share_vector_from_seed,
       prng->GenerateUniformRandomRange(input.size(), param.modulus()));
 
+  ASSERT_EQ(secret_share.share_vector().size(), input.size());
+  ASSERT_EQ(share_vector_from_seed.size(), input.size());
+
   for (int i = 0; i < input.size(); i++) {
     ASSERT_EQ(input[i],
               (share_vector_from_seed[i] + secret_share.share_vector(i)) %
@@ -96,6 +99,9 @@ TEST(AdditiveSecretSharing, SecretShareOverPrimeFieldElementsSucceeds) {
       std::vector<uint32_t> share_vector_from_seed,
       prng->GenerateUniformRandomRange(input.size(), param.modulus()));
 
+  ASSERT_EQ(secret_share.share_vector().size(), input.size());
+  ASSERT_EQ(share_vector_from_seed.size(), input.size());
+
   for (int i = 0; i < input.size(); i++) {
     ASSERT_EQ(input[i],
               (share_vector_from_seed[i] + secret_share.share_vector(i)) %