add a test for the extension array adder.

Lokathor · Sep 1, 2023 · 7162af0 · 7162af0
1 parent 234854b
commit 7162af0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/pcg.rs b/src/pcg.rs
@@ -193,3 +193,17 @@ impl<const K: usize> PCG32K<K> {
     }
   }
 }
+
+#[test]
+fn test_ext_add() {
+  let mut x = PCG32K::<2> { state: 0, ext: [u32::MAX, 0] };
+  x.ext_add(1);
+  assert_eq!(x.ext[0], 0);
+  assert_eq!(x.ext[1], 1);
+  //
+  let mut x = PCG32K::<3> { state: 0, ext: [u32::MAX, u32::MAX, 0] };
+  x.ext_add(1);
+  assert_eq!(x.ext[0], 0);
+  assert_eq!(x.ext[1], 0);
+  assert_eq!(x.ext[2], 1);
+}