improve coverage

pyproject.toml

@@ -13,8 +13,19 @@ addopts = [
     "--ignore setup.py",
     "--ignore run_test.py",
     "--cov-report term-missing",
+    "--cov-branch",
     "--tb native",
     "--strict-markers",
     "--durations=20",
 ]
 markers = ["serial: execute test serially (to avoid race conditions)"]
+
+[tool.coverage.run]
+source = [ "src/", ]
+omit = [
+    "setup.py",
+    "src/conda_package_handling/__main__.py",
+    "src/conda_package_handling/_version.py",
+    "versioneer.py",
+    "tests/*",
+]

src/conda_package_handling/api.py

@@ -11,7 +11,6 @@
 from .tarball import CondaTarBZ2 as _CondaTarBZ2
 from .utils import ensure_list, filter_info_files
 from .utils import get_executor as _get_executor
-from .utils import rm_rf as _rm_rf
 
 SUPPORTED_EXTENSIONS: dict[str, type[AbstractBaseFormat]] = {".tar.bz2": _CondaTarBZ2}
 
@@ -113,7 +112,7 @@ def create(prefix, file_list, out_fn, out_folder=None, **kw):
                 # don't leave broken files around
                 abs_out_fn = _os.path.join(out_folder, out_fn)
                 if _os.path.isfile(abs_out_fn):
-                    _rm_rf(abs_out_fn)
+                    _os.unlink(abs_out_fn)
                 raise err
     else:
         raise ValueError(
@@ -191,7 +190,7 @@ def is_info(filename):
         except BaseException as e:
             # don't leave partial package around
             if _os.path.isfile(out_fn):
-                _rm_rf(out_fn)
+                _os.unlink(out_fn)
             if not isinstance(e, Exception):
                 raise
             errors = str(e)
@@ -216,7 +215,10 @@ def transmute(in_file, out_ext, out_folder=None, processes=1, **kw):
         for fn, out_fn, errors in executor.map(convert_f, flist):
             if errors:
                 failed_files[fn] = errors
-                _rm_rf(out_fn)
+                try:
+                    _os.unlink(out_fn)
+                except FileNotFoundError:
+                    pass
     return failed_files
 
 

src/conda_package_handling/streaming.py

@@ -7,6 +7,7 @@
 import io
 import tarfile
 from contextlib import redirect_stdout
+from pathlib import Path
 from tarfile import TarError
 from typing import Generator
 from zipfile import BadZipFile
@@ -18,9 +19,9 @@
 
 
 def _stream_components(
-    filename: str,
+    filename: str | Path,
     components: list[str],
-    dest_dir: str = "",
+    dest_dir: str | Path = "",
 ) -> Generator[Generator[tuple[tarfile.TarFile, tarfile.TarInfo]]]:
     if str(filename).endswith(".tar.bz2"):
         assert components == ["pkg"]
@@ -38,7 +39,7 @@ def _stream_components(
         raise exceptions.InvalidArchiveError(filename, f"failed with error: {str(e)}") from e
 
 
-def _extract(filename: str, dest_dir: str, components: list[str]):
+def _extract(filename: str | Path, dest_dir: str | Path, components: list[str]):
     """
     Extract .conda or .tar.bz2 package to dest_dir.