aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/Lib/test/test_pathlib/test_write.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_pathlib/test_write.py')
-rw-r--r--Lib/test/test_pathlib/test_write.py21
1 files changed, 14 insertions, 7 deletions
diff --git a/Lib/test/test_pathlib/test_write.py b/Lib/test/test_pathlib/test_write.py
index 3d6057fb9dd..040af7be152 100644
--- a/Lib/test/test_pathlib/test_write.py
+++ b/Lib/test/test_pathlib/test_write.py
@@ -6,12 +6,16 @@ import io
import os
import unittest
-from pathlib import Path
-from pathlib.types import _WritablePath
-from pathlib._os import magic_open
+from .support import is_pypi
+from .support.local_path import WritableLocalPath, LocalPathGround
+from .support.zip_path import WritableZipPath, ZipPathGround
-from test.test_pathlib.support.local_path import WritableLocalPath, LocalPathGround
-from test.test_pathlib.support.zip_path import WritableZipPath, ZipPathGround
+if is_pypi:
+ from pathlib_abc import _WritablePath
+ from pathlib_abc._os import magic_open
+else:
+ from pathlib.types import _WritablePath
+ from pathlib._os import magic_open
class WriteTestBase:
@@ -101,8 +105,11 @@ class LocalPathWriteTest(WriteTestBase, unittest.TestCase):
ground = LocalPathGround(WritableLocalPath)
-class PathWriteTest(WriteTestBase, unittest.TestCase):
- ground = LocalPathGround(Path)
+if not is_pypi:
+ from pathlib import Path
+
+ class PathWriteTest(WriteTestBase, unittest.TestCase):
+ ground = LocalPathGround(Path)
if __name__ == "__main__":