diff options
Diffstat (limited to 'Lib/test/test_pathlib/test_read.py')
-rw-r--r-- | Lib/test/test_pathlib/test_read.py | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/Lib/test/test_pathlib/test_read.py b/Lib/test/test_pathlib/test_read.py index 49015dac3b3..938d3a1e987 100644 --- a/Lib/test/test_pathlib/test_read.py +++ b/Lib/test/test_pathlib/test_read.py @@ -6,12 +6,16 @@ import collections.abc import io import unittest -from pathlib import Path -from pathlib.types import PathInfo, _ReadablePath -from pathlib._os import magic_open +from .support import is_pypi +from .support.local_path import ReadableLocalPath, LocalPathGround +from .support.zip_path import ReadableZipPath, ZipPathGround -from test.test_pathlib.support.local_path import ReadableLocalPath, LocalPathGround -from test.test_pathlib.support.zip_path import ReadableZipPath, ZipPathGround +if is_pypi: + from pathlib_abc import PathInfo, _ReadablePath + from pathlib_abc._os import magic_open +else: + from pathlib.types import PathInfo, _ReadablePath + from pathlib._os import magic_open class ReadTestBase: @@ -301,8 +305,11 @@ class LocalPathReadTest(ReadTestBase, unittest.TestCase): ground = LocalPathGround(ReadableLocalPath) -class PathReadTest(ReadTestBase, unittest.TestCase): - ground = LocalPathGround(Path) +if not is_pypi: + from pathlib import Path + + class PathReadTest(ReadTestBase, unittest.TestCase): + ground = LocalPathGround(Path) if __name__ == "__main__": |