Cast root to str to allow e.g. a pathlib.Path object to be used (…

…handley-lab#358)

* cast root to string in case it is a path

* move import to top

* bump version to 2.6.1

---------

Co-authored-by: Will Handley <[email protected]>

README.rst

@@ -2,7 +2,7 @@
 anesthetic: nested sampling post-processing
 ===========================================
 :Authors: Will Handley and Lukas Hergt
-:Version: 2.7.2
+:Version: 2.7.3
 :Homepage: https://github.com/handley-lab/anesthetic
 :Documentation: http://anesthetic.readthedocs.io/
 

anesthetic/_version.py

@@ -1 +1 @@
-__version__ = '2.7.2'
+__version__ = '2.7.3'

anesthetic/read/chain.py

@@ -26,7 +26,7 @@ def read_chains(root, *args, **kwargs):
 
     Parameters
     ----------
-    root : str
+    root : str, pathlib.Path
         root name for reading files
 
     *args, **kwargs:
@@ -39,6 +39,7 @@ def read_chains(root, *args, **kwargs):
     :class:`anesthetic.samples.MCMCSamples` depending on auto-detection
 
     """
+    root = str(root)
     if 'burn_in' in kwargs:
         raise KeyError(
             "This is anesthetic 1.0 syntax. The `burn_in` keyword is no "

tests/test_reader.py

@@ -1,5 +1,6 @@
 import anesthetic.examples._matplotlib_agg  # noqa: F401
 import os
+from pathlib import Path
 import pytest
 import numpy as np
 from numpy.testing import assert_array_equal, assert_array_almost_equal
@@ -300,3 +301,9 @@ def test_hdf5(tmp_path, root):
     samples_ = read_hdf(filename, key)
     assert_frame_equal(samples_, samples)
     assert type(samples_) is type(samples)
+
+
+@pytest.mark.parametrize('root', ['pc', 'gd'])
+def test_path(root):
+    base_dir = Path("./tests/example_data")
+    read_chains(base_dir / root)