fix: add SecretInteger test

NillionNetwork · Sep 2, 2024 · df188f0 · df188f0
1 parent f208f2b
commit df188f0
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/tests/nada-tests/src/shuffle.py b/tests/nada-tests/src/shuffle.py
@@ -16,17 +16,21 @@ def nada_main() -> List[Output]:
     a = na.array([8], parties[0], "A", na.Rational)
     b = na.array([8], parties[0], "B", na.SecretRational)
     c = na.array([8], parties[0], "C", PublicInteger)
+    d = na.array([8], parties[0], "D", SecretInteger)
 
     shuffled_a = shuffle(a)
     shuffled_b = shuffle(b)
     shuffled_c = shuffle(c)
+    shuffled_d = shuffle(d)
 
     result_a = shuffled_a - shuffled_a
     result_b = shuffled_b - shuffled_b
     result_c = shuffled_c - shuffled_c
+    result_d = shuffled_d - shuffled_d
 
     return (
         na.output(result_a, parties[1], "my_output_a")
         + na.output(result_b, parties[1], "my_output_b")
         + na.output(result_c, parties[1], "my_output_c")
+        + na.output(result_d, parties[1], "my_output_d")
     )
diff --git a/tests/nada-tests/tests/shuffle.yaml b/tests/nada-tests/tests/shuffle.yaml
@@ -25,6 +25,14 @@ inputs:
   A_6: 6
   A_4: 4
   B_5: 5
+  D_0: 0
+  D_1: 1
+  D_2: 2
+  D_3: 3
+  D_4: 4
+  D_5: 5
+  D_6: 6
+  D_7: 7
 expected_outputs:
   my_output_a_0: 0
   my_output_b_2: 0
@@ -50,3 +58,11 @@ expected_outputs:
   my_output_a_3: 0
   my_output_b_1: 0
   my_output_a_5: 0
+  my_output_d_0: 0
+  my_output_d_1: 0
+  my_output_d_2: 0
+  my_output_d_3: 0
+  my_output_d_4: 0
+  my_output_d_5: 0
+  my_output_d_6: 0
+  my_output_d_7: 0