Add a test

PyO3 · Nov 29, 2022 · 79501bc · 79501bc
1 parent 850cc93
commit 79501bc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/once_cell.rs b/src/once_cell.rs
@@ -212,4 +212,21 @@ mod tests {
             assert_eq!(dict.get_item(foo3).unwrap().extract::<usize>().unwrap(), 42);
         });
     }
+
+    #[test]
+    fn test_once_cell() {
+        Python::with_gil(|py| {
+            let cell = GILOnceCell::new();
+
+            assert!(cell.get(py).is_none());
+
+            assert_eq!(cell.get_or_try_init(py, || Err(5)), Err(5));
+            assert!(cell.get(py).is_none());
+
+            assert_eq!(cell.get_or_try_init(py, || Ok::<_, ()>(2)), Ok(&2));
+            assert_eq!(cell.get(py), Some(&2));
+
+            assert_eq!(cell.get_or_try_init(py, || Err(5)), Ok(&2));
+        })
+    }
 }