python: Install data files as package data

Author: dalcinl

.github/workflows/test_python.yml

@@ -27,7 +27,4 @@ jobs:
           run: python -m pip install ".[test]"
 
         - name: Run tests
-          run: >
-            export MPP_DIRECTORY=$(pwd) &&
-            export MPP_DATA_DIRECTORY=$MPP_DIRECTORY/data &&
-            pytest
+          run: pytest

.github/workflows/wheels.yml

@@ -20,10 +20,7 @@ jobs:
           run: python -m pip install ".[test]"
 
         - name: Run tests
-          run: >
-            export MPP_DIRECTORY=$(pwd) &&
-            export MPP_DATA_DIRECTORY=$MPP_DIRECTORY/data &&
-            pytest
+          run: pytest
 
     test_mac:
       runs-on: macos-latest
@@ -40,10 +37,7 @@ jobs:
           run: python -m pip install ".[test]"
 
         - name: Run tests
-          run: >
-            export MPP_DIRECTORY=$(pwd) &&
-            export MPP_DATA_DIRECTORY=$MPP_DIRECTORY/data &&
-            pytest
+          run: pytest
 
     linux_wheels:
         strategy:

interface/python/CMakeLists.txt

@@ -14,3 +14,5 @@ target_link_libraries(_mutationpp PRIVATE mutation++)
 install(DIRECTORY mutationpp DESTINATION .)
 
 install(TARGETS _mutationpp DESTINATION mutationpp)
+
+install(DIRECTORY "${CMAKE_SOURCE_DIR}/data" DESTINATION mutationpp)

interface/python/src/pyGlobalOptions.cpp

@@ -7,8 +7,26 @@ namespace py = pybind11;
  * Python wrapper definition for the GlobalOptions class.
  */
 
+namespace {
+
+  inline void setDefaultDataDirectory(const std::string& datadir) {
+    if (std::getenv("MPP_DATA_DIRECTORY") == nullptr) {
+      Mutation::GlobalOptions::dataDirectory(datadir);
+    }
+  };
+
+}
+
 void py_export_GlobalOptions(py::module &m) {
 
+  const py::object resources = py::module_::import("importlib.resources");
+  const py::object pkgname = m.attr("__spec__").attr("parent");
+  const py::object rootdir = resources.attr("files")(pkgname);
+  const py::object datadir = rootdir / py::str("data");
+  const std::string data_directory = py::str(datadir).cast<std::string>();
+
+  setDefaultDataDirectory(data_directory);
+
   /**
    * Overloaded member functions wrappers
    */
@@ -29,8 +47,9 @@ void py_export_GlobalOptions(py::module &m) {
       .def_static("workingDirectory", []() {
         return Mutation::GlobalOptions::workingDirectory();
       })
-      .def_static("reset", []() {
+      .def_static("reset", [data_directory]() {
         Mutation::GlobalOptions::reset();
+        setDefaultDataDirectory(data_directory);
       })
     ;
 }

tests/python/test_options.py

@@ -4,7 +4,8 @@
 import unittest.mock as mock
 import mutationpp as mpp
 
-INITIAL_DATADIR = os.environ.get("MPP_DATA_DIRECTORY", "")
+PACKAGE_DATADIR = os.fspath(importlib.resources.files(mpp) / "data")
+INITIAL_DATADIR = os.environ.get("MPP_DATA_DIRECTORY", PACKAGE_DATADIR)
 
 
 def test_options_datadir_initial():
@@ -62,6 +63,6 @@ def test_options_reset(datadir):
     with mock.patch.dict(os.environ, clear=True):
         mpp.GlobalOptions.reset()
         set_datadir = mpp.GlobalOptions.dataDirectory()
-        assert set_datadir == ""
+        assert set_datadir == PACKAGE_DATADIR
         set_workdir = mpp.GlobalOptions.workingDirectory()
         assert set_workdir == ""