[2024-12-16] Changing to mod syntax in Rust

src/lib.rs

@@ -1,10 +1,14 @@
 use pyo3::prelude::*;
-use std::time::Instant;
-use pyo3::exceptions::PyException;
 
-#[pyclass]
-#[derive(Clone)]
-struct Results {
+#[pymodule]
+mod tictoc {
+    use super::*;
+    use std::time::Instant;
+    use pyo3::exceptions::PyException;
+    
+    #[pyclass]
+    #[derive(Clone)]
+    struct Results {
         #[pyo3(get)]
         nanos: u128,
         #[pyo3(get)]
@@ -13,18 +17,18 @@ struct Results {
         millis: u128,
         #[pyo3(get)]
         seconds: f64,
-}
+    }
     
-#[pyclass(module = "tictoc", name = "init")]
-struct Init {
+    #[pyclass(name = "init")]
+    pub struct Init {
         time: Instant,
         #[pyo3(get)]
         results: Results,
         status: bool,
-}
+    }
     
-#[pymethods]
-impl Init {
+    #[pymethods]
+    impl Init {
         #[new]
         fn new() -> Self {
             let res = Results {
@@ -59,34 +63,28 @@ impl Init {
                 Ok(())
             }
         }
-}
-
-#[pymodule]
-fn tictoc(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
-    m.add_class::<Init>()?;
-    Ok(())
-}
-
-#[test]
-fn test_new() {
+    }
+    #[test]
+    fn test_new() {
         let init = Init::new();
         assert_eq!(init.results.nanos,0);
-}
+    }
     
-#[test]
-fn test_tic() {
+    #[test]
+    fn test_tic() {
         let mut init = Init::new();
         let time1 = init.time;
         init.tic();
         let time2 = init.time;
         assert!(time2 > time1)
-}
+    }
     
-#[test]
-fn test_toc() {
+    #[test]
+    fn test_toc() {
         let mut init = Init::new();
         init.tic();
         println!("{}","test");
         let _ = init.toc();
         assert!(init.results.nanos > 0)
+    }
 }