1 files changed, 35 insertions, 39 deletions

diff --git a/CLI/rust/src/game_of_life/simulation/plane.rs b/CLI/rust/src/game_of_life/simulation/plane.rs
index fad9398..76d91fb 100644
--- a/CLI/rust/src/game_of_life/simulation/plane.rs
+++ b/CLI/rust/src/game_of_life/simulation/plane.rs
@@ -16,6 +16,24 @@ impl Index<isize> for Plane {
     }
 }
 
+#[cfg(test)]
+mod index {
+    use super::*;
+
+    #[test]
+    fn get_element() {
+        let p = Plane(vec![false, false, true, false, false]);
+        assert_eq!(p[0], false);
+        assert_eq!(p[1], false);
+        assert_eq!(p[3], false);
+        assert_eq!(p[4], false);
+
+        assert_eq!(p[-3], true);
+        assert_eq!(p[2], true);
+        assert_eq!(p[7], true);
+    }
+}
+
 impl IndexMut<isize> for Plane {
     fn index_mut(&mut self, idx: isize) -> &mut Self::Output {
         let len = self.0.len() as isize;
@@ -24,49 +42,27 @@ impl IndexMut<isize> for Plane {
 }
 
 #[cfg(test)]
-mod plane {
+mod index_mut {
     use super::*;
 
-    mod index {
-        use super::*;
-
-        #[test]
-        fn get_element() {
-            let p = Plane(vec![false, false, true, false, false]);
-            assert_eq!(p[0], false);
-            assert_eq!(p[1], false);
-            assert_eq!(p[3], false);
-            assert_eq!(p[4], false);
-
-            assert_eq!(p[-3], true);
-            assert_eq!(p[2], true);
-            assert_eq!(p[7], true);
-        }
+    #[test]
+    fn set_negative_index_element() {
+        let mut p = Plane(vec![false, false, false, false, false]);
+        p[-3] = true;
+        assert_eq!(p, Plane(vec![false, false, true, false, false]));
     }
 
-    #[cfg(test)]
-    mod index_mut {
-        use super::*;
-
-        #[test]
-        fn set_negative_index_element() {
-            let mut p = Plane(vec![false, false, false, false, false]);
-            p[-3] = true;
-            assert_eq!(p, Plane(vec![false, false, true, false, false]));
-        }
-
-        #[test]
-        fn set_element() {
-            let mut p = Plane(vec![false, false, false, false, false]);
-            p[2] = true;
-            assert_eq!(p, Plane(vec![false, false, true, false, false]));
-        }
+    #[test]
+    fn set_element() {
+        let mut p = Plane(vec![false, false, false, false, false]);
+        p[2] = true;
+        assert_eq!(p, Plane(vec![false, false, true, false, false]));
+    }
 
-        #[test]
-        fn set_out_of_bound_element() {
-            let mut p = Plane(vec![false, false, false, false, false]);
-            p[7] = true;
-            assert_eq!(p, Plane(vec![false, false, true, false, false]));
-        }
+    #[test]
+    fn set_out_of_bound_element() {
+        let mut p = Plane(vec![false, false, false, false, false]);
+        p[7] = true;
+        assert_eq!(p, Plane(vec![false, false, true, false, false]));
     }
 }