test: improve coverage

Author: dvp2015

src/mapstp/merge.py

@@ -32,7 +32,7 @@ def is_defined(number: Union[int, float]) -> bool:
         true - if `number` is a valid number,
         false - otherwise
     """
-    return number is not None and not math.isnan(number)
+    return number is not None and number is not pd.NA and not math.isnan(number)
 
 
 def extract_number_and_density(

src/mapstp/utils/resource.py

@@ -61,6 +61,6 @@ def _resolve_package(package: str = None) -> str:
     if package is None:
         module = inspect.getmodule(inspect.stack()[2][0])
         if module is None:
-            raise ValueError("Cannot define package.")
+            raise ValueError("Cannot define package.")  # pragma: no cover
         package = module.__name__
     return package

tests/test_merge.py

@@ -1,4 +1,6 @@
 import mapstp.merge as m
+import numpy as np
+import pandas as pd
 import pytest
 
 
@@ -7,5 +9,37 @@ def test_merger(data):
     assert sections.cells is not None
 
 
+@pytest.mark.parametrize(
+    "number, density, factor, expected",
+    [
+        (1, 7.93, pd.NA, (1, 7.93)),
+        (1, 7.93, np.NAN, (1, 7.93)),
+        (1, 2.0, 2.0, (1, 4.0)),
+    ],
+)
+def test_extract_number_and_density(number, density, factor, expected):
+    df = pd.DataFrame.from_records(
+        data=[(number, density, factor)], columns="number density factor".split()
+    )
+    actual = m.extract_number_and_density(0, df)
+    assert expected == actual
+
+
+@pytest.mark.parametrize(
+    "number, density, factor, exception",
+    [
+        (-1, 7.93, pd.NA, m.PathInfoError),
+        (1, -7.93, np.NAN, m.PathInfoError),
+        (1, 2.0, -2.0, m.PathInfoError),
+    ],
+)
+def test_extract_number_and_density_bad_path(number, density, factor, exception):
+    df = pd.DataFrame.from_records(
+        data=[(number, density, factor)], columns="number density factor".split()
+    )
+    with pytest.raises(exception):
+        m.extract_number_and_density(0, df)
+
+
 if __name__ == "__main__":
     pytest.main()

tests/utils/test_resource.py

@@ -39,6 +39,12 @@ def test_filename_resolver_when_resource_doesnt_exist(package, resource, expecte
     ).exists(), f"The resource '{resource}' should not be available"
 
 
+def test_filename_resolver_when_package_doesnt_exist():
+    resolver = filename_resolver("not_existing")
+    with pytest.raises(ModuleNotFoundError):
+        resolver("something.txt")
+
+
 def test_path_resolver():
     resolver = path_resolver()
     actual = resolver(THIS_FILENAME)