Rework joinpath for importlib.resources

1 files changed, 40 insertions, 0 deletions

diff --git a/library/python/runtime_py3/test/test_resources.py b/library/python/runtime_py3/test/test_resources.py
index 059cc039e6..75c1eb3549 100644
--- a/library/python/runtime_py3/test/test_resources.py
+++ b/library/python/runtime_py3/test/test_resources.py
@@ -71,3 +71,43 @@ def test_read_text_missing():
 )
 def test_contents_good_path(package, expected):
     assert sorted(ir.contents(package)) == sorted(expected)
+
+
+def test_files_joinpath():
+    assert ir.files("resources") / "submodule"
+    assert ir.files("resources") / "foo.txt"
+    assert ir.files("resources") / "submodule" / "bar.txt"
+    assert ir.files("resources.submodule") / "bar.txt"
+
+
+@pytest.mark.parametrize(
+    "package, resource, expected",
+    (
+        ("resources", "foo.txt", b"bar"),
+        ("resources.submodule", "bar.txt", b"foo"),
+    ),
+)
+def test_files_read_bytes(package, resource, expected):
+    assert (ir.files(package) / resource).read_bytes() == expected
+
+
+@pytest.mark.parametrize(
+    "package, resource, expected",
+    (
+        ("resources", "foo.txt", "bar"),
+        ("resources.submodule", "bar.txt", "foo"),
+    ),
+)
+def test_files_read_text(package, resource, expected):
+    assert (ir.files(package) / resource).read_text() == expected
+
+
+@pytest.mark.parametrize(
+    "package, expected",
+    (
+        ("resources", ("foo.txt", "submodule")),
+        ("resources.submodule", ("bar.txt",)),
+    ),
+)
+def test_files_iterdir(package, expected):
+    assert tuple(resource.name for resource in ir.files(package).iterdir()) == expected