fix: wren core raise CoreError instead of pyo3_runtime.PanicException

wren-core-py/src/extractor.rs

@@ -16,10 +16,10 @@ pub struct PyExtractor {
 #[pymethods]
 impl PyExtractor {
     #[new]
-    pub fn new(mdl_base64: &str) -> Self {
-        let manifest = to_manifest(mdl_base64).unwrap();
+    pub fn new(mdl_base64: &str) -> Result<Self, CoreError> {
+        let manifest = to_manifest(mdl_base64)?;
         let mdl = WrenMDL::new_ref(manifest);
-        Self { mdl }
+        Ok(Self { mdl })
     }
 
     /// parse the given SQL and return the list of used table name.

wren-core-py/tests/test_modeling_core.py

@@ -142,6 +142,29 @@ def test_get_available_functions():
     assert max_if["param_types"] is None
 
 
+@pytest.mark.parametrize(
+    ("value", "expected_error", "error_message"),
+    [
+        (
+            None,
+            TypeError,
+            "argument 'mdl_base64': 'NoneType' object cannot be converted to 'PyString'",
+        ),
+        ("xxx", Exception, "Base64 decode error: Invalid padding"),
+        ("{}", Exception, "Base64 decode error: Invalid symbol 123, offset 0."),
+        (
+            "",
+            Exception,
+            "Serde JSON error: EOF while parsing a value at line 1 column 0",
+        ),
+    ],
+)
+def test_extractor_with_invalid_manifest(value, expected_error, error_message):
+    with pytest.raises(expected_error) as e:
+        Extractor(value)
+    assert str(e.value) == error_message
+
+
 @pytest.mark.parametrize(
     ("sql", "expected"),
     [