Pathlib object handling for Universe, SingleFrameReaderBase and Toplogy parsers (Issue #3937)

package/CHANGELOG

@@ -16,8 +16,9 @@ The rules for this file:
 -------------------------------------------------------------------------------
 ??/??/?? IAlibay, HeetVekariya, marinegor, lilyminium, RMeli,
          ljwoods2, aditya292002, pstaerk, PicoCentauri, BFedder,
-         tyler.je.reddy, SampurnaM, leonwehrhan, kainszs, orionarcher
-
+         tyler.je.reddy, SampurnaM, leonwehrhan, kainszs, orionarcher,
+         talagayev, hmacdope
+
  * 2.8.0
 
 Fixes
@@ -39,6 +40,7 @@ Fixes
  * Fix groups.py doctests using sphinx directives (Issue #3925, PR #4374)
 
 Enhancements
+ * Handling of Pathlib.Path in SingleFrameReaderBase, Topology and Universe (Issue #3937)
  * Added a tqdm progress bar for `MDAnalysis.analysis.pca.PCA.transform()`
    (PR #4531)
  * Improved performance of PDBWriter (Issue #2785, PR #4472)

package/MDAnalysis/coordinates/base.py

@@ -1663,7 +1663,11 @@ class SingleFrameReaderBase(ProtoReader):
     def __init__(self, filename, convert_units=True, n_atoms=None, **kwargs):
         super(SingleFrameReaderBase, self).__init__()
 
-        self.filename = filename
+        if isinstance(filename, NamedStream):
+            self.filename = filename
+        else:
+            self.filename = str(filename)
+
         self.convert_units = convert_units
 
         self.n_frames = 1

package/MDAnalysis/core/universe.py

@@ -60,6 +60,7 @@
 import warnings
 import contextlib
 import collections
+import pathlib
 
 import MDAnalysis
 import sys
@@ -99,6 +100,9 @@ def _check_file_like(topology):
         else:
             _name = None
         return NamedStream(topology, _name)
+
+    elif isinstance(topology, pathlib.Path):
+        return str(topology)
     return topology
 
 def _topology_from_file_like(topology_file, topology_format=None,

package/MDAnalysis/topology/base.py

@@ -47,6 +47,7 @@
 from ..lib import util
 
 
+
 class _Topologymeta(type):
     """Internal: Topology Parser registration voodoo
 
@@ -115,7 +116,12 @@ class TopologyReaderBase(IOBase, metaclass=_Topologymeta):
        Added keyword 'universe' to pass to Atom creation.
     """
     def __init__(self, filename):
-        self.filename = filename
+
+        if isinstance(filename, util.NamedStream):
+            self.filename = filename
+        else:
+            self.filename = str(filename)
+
 
     def parse(self, **kwargs):  # pragma: no cover
         raise NotImplementedError("Override this in each subclass")

testsuite/MDAnalysisTests/coordinates/base.py

@@ -25,12 +25,14 @@
 
 import numpy as np
 import pytest
+from pathlib import Path
 from unittest import TestCase
 from numpy.testing import (assert_equal, assert_almost_equal,
                            assert_array_almost_equal, assert_allclose)
 
 import MDAnalysis as mda
 from MDAnalysis.coordinates.timestep import Timestep
+from MDAnalysis.coordinates.memory import MemoryReader
 from MDAnalysis.transformations import translate
 
 
@@ -126,6 +128,12 @@ def test_pickle_singleframe_reader(self):
         assert_equal(reader.ts, reader_p.ts,
                      "Single-frame timestep is changed after pickling")
 
+    def test_pathlib_input_single(self):
+        path = Path(self.filename)
+        u_str = mda.Universe(self.filename)
+        u_path = mda.Universe(path)
+        assert u_str.atoms.n_atoms == u_path.atoms.n_atoms
+
 
 class BaseReference(object):
     def __init__(self):
@@ -521,6 +529,16 @@ def test_timeseries_atomgroup_asel_mutex(self, reader):
         atoms = mda.Universe(reader.filename).select_atoms("index 1")
         with pytest.raises(ValueError, match="Cannot provide both"):
             timeseries = reader.timeseries(atomgroup=atoms, asel=atoms, order='fac')
+
+    def test_pathlib_input_base(self, reader):     
+        if isinstance(reader, MemoryReader):
+            if isinstance(reader, MemoryReader):
+                skip_reason = "MemoryReader"
+            pytest.skip(f"Skipping test for Pathlib input with reason: {skip_reason}")
+        path = Path(reader.filename)
+        u_str = mda.Universe(reader.filename)
+        u_path = mda.Universe(path)
+        assert u_str.atoms.n_atoms == u_path.atoms.n_atoms
 
 
 class MultiframeReaderTest(BaseReaderTest):

testsuite/MDAnalysisTests/coordinates/test_gro.py

@@ -46,7 +46,6 @@
     assert_equal,
 )
 import pytest
-from io import StringIO
 
 
 class TestGROReaderOld(RefAdK):

testsuite/MDAnalysisTests/topology/base.py

@@ -100,3 +100,11 @@ def test_creates_universe(self, filename):
         """Check that Universe works with this Parser"""
         u = mda.Universe(filename)
         assert isinstance(u, mda.Universe)
+
+    def test_pathlib_input(self, filename):
+        """Check that pathlib.Path objects are accepted"""
+        import pathlib
+        path = pathlib.Path(filename)
+        u_str = mda.Universe(filename)
+        u_path = mda.Universe(path)
+        assert u_str.atoms.n_atoms == u_path.atoms.n_atoms