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

src/lib.rs

@@ -1,4 +1,8 @@
 use pyo3::prelude::*;
+
+#[pymodule]
+mod tictoc {
+    use super::*;
     use std::time::Instant;
     use pyo3::exceptions::PyException;
     
@@ -15,8 +19,8 @@ struct Results {
         seconds: f64,
     }
     
-#[pyclass(module = "tictoc", name = "init")]
+    #[pyclass(name = "init")]
-struct Init {
+    pub struct Init {
         time: Instant,
         #[pyo3(get)]
         results: Results,
@@ -60,13 +64,6 @@ impl Init {
             }
         }
     }
-
-#[pymodule]
-fn tictoc(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
-    m.add_class::<Init>()?;
-    Ok(())
-}
-
     #[test]
     fn test_new() {
         let init = Init::new();
@@ -90,3 +87,4 @@ fn test_toc() {
         let _ = init.toc();
         assert!(init.results.nanos > 0)
     }
+}