File tree Expand file tree Collapse file tree 1 file changed +45
-0
lines changed
importlib_resources/tests Expand file tree Collapse file tree 1 file changed +45
-0
lines changed Original file line number Diff line number Diff line change
1
+ import unittest
2
+ import contextlib
3
+ import pathlib
4
+
5
+ import importlib_resources as resources
6
+ from ..abc import TraversableResources , ResourceReader
7
+ from . import util
8
+ from ._compat import os_helper
9
+
10
+
11
+ class SimpleLoader :
12
+ """
13
+ A simple loader that only implements a resource reader.
14
+ """
15
+
16
+ def __init__ (self , reader : ResourceReader ):
17
+ self .reader = reader
18
+
19
+ def get_resource_reader (self , package ):
20
+ return self .reader
21
+
22
+
23
+ class MagicResources (TraversableResources ):
24
+ """
25
+ Magically returns the resources at path.
26
+ """
27
+
28
+ def __init__ (self , path : pathlib .Path ):
29
+ self .path = path
30
+
31
+ def files (self ):
32
+ return self .path
33
+
34
+
35
+ class CustomTraversableResourcesTests (unittest .TestCase ):
36
+ def setUp (self ):
37
+ self .fixtures = contextlib .ExitStack ()
38
+ self .addCleanup (self .fixtures .close )
39
+
40
+ def test_custom_loader (self ):
41
+ temp_dir = self .fixtures .enter_context (os_helper .temp_dir ())
42
+ loader = SimpleLoader (MagicResources (temp_dir ))
43
+ pkg = util .create_package_from_loader (loader )
44
+ files = resources .files (pkg )
45
+ assert files is temp_dir
You can’t perform that action at this time.
0 commit comments