Basic tests

1 files changed, 29 insertions, 8 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 2f49852..a6cd2ef 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,17 +1,30 @@
 
 pub struct BCH {
-    n: u32,
+    m: u32,
 }
 
 impl BCH {
-    pub fn new(n: u32) -> BCH {
-        BCH {
-            n
+    pub fn new(m: u32, _t: u32, _prim_poly: u32) -> Option<BCH> {
+        let _err: u32 = 0;
+        let _i: u32;
+        let _words: u32;
+        let _genpoly: u32;
+
+        const MIN_M: u32 = 5;
+        const MAX_M: u32 = 15;
+
+        if (m < MIN_M) || (m > MAX_M) {
+            // `m` values must be between5 and 15 in this implementation
+            return None
         }
+
+        Some(BCH {
+            m
+        })
     }
 
     pub fn decode(self) -> u32{
-        return self.n
+        return self.m
     }
 }
 
@@ -20,8 +33,16 @@ mod tests {
     use super::*;
 
     #[test]
-    fn it_works() {
-        let bch = BCH::new(3);
-        assert_eq!(bch.decode(), 3);
+    fn test_bch_init() {
+        let bch = BCH::new(6, 0, 0).unwrap();
+        assert_eq!(bch.decode(), 6);
+    }
+
+    #[test]
+    fn fail_m_values() {
+        let bch = BCH::new(4, 0, 0);
+        assert_eq!(bch.is_none(), true);
+        let bch = BCH::new(16, 0, 0);
+        assert_eq!(bch.is_none(), true);
     }
 }