wrap pyvcell-fvsolver in pyvcell.solvers.fvsolver with safe typing

pyvcell/solvers/fvsolver.py

@@ -0,0 +1,19 @@
+from pathlib import Path
+
+from pyvcell_fvsolver import solve as _fv_solve, version as _fv_version  # type: ignore[import-untyped]
+
+
+def solve(input_file: Path | str, vcg_file: Path | str, output_dir: Path | str) -> int:
+    return_code = _fv_solve(fvInputFilename=str(input_file), vcgInputFilename=str(vcg_file), outputDir=str(output_dir))
+    if not isinstance(return_code, int):
+        raise TypeError(f"Expected int but got {type(return_code)}")
+    if return_code != 0:
+        raise ValueError(f"Error in solve: {return_code}")
+    return return_code
+
+
+def version() -> str:
+    version_value = _fv_version()
+    if not isinstance(version_value, str):
+        raise TypeError(f"Expected str but got {type(version_value)}")
+    return version_value

tests/fvsolver_test.py

@@ -1,7 +1,7 @@
 import os
 from pathlib import Path
 
-from pyvcell_fvsolver import __version__, solve, version  # type: ignore[import-untyped]
+from pyvcell.solvers.fvsolver import solve, version
 
 # get parent directory of this script as a path
 parent_dir: Path = Path(os.path.dirname(os.path.realpath(__file__))).parent
@@ -12,10 +12,6 @@
 test_output_dir_2 = parent_dir / "test_output_2"
 
 
-def test_version_var() -> None:
-    assert __version__ == "0.0.4"
-
-
 def test_version_func() -> None:
     assert version() is not None
 
@@ -27,7 +23,8 @@ def test_solve() -> None:
     for file in test_output_dir_2.iterdir() if test_output_dir_2.exists() else []:
         file.unlink()
 
-    retcode_1: int = solve(
-        fvInputFilename=str(fv_input_file), vcgInputFilename=str(vcg_input_file), outputDir=str(test_output_dir_1)
-    )
+    retcode_1: int = solve(input_file=fv_input_file, vcg_file=vcg_input_file, output_dir=test_output_dir_1)
+    assert test_output_dir_1.exists()
+    assert len(list(test_output_dir_1.iterdir())) > 0
+
     print(f"retcode_1: {retcode_1}")