Test zip data for read()

python · Oct 27, 2017 · ca80944 · ca80944
1 parent 6daf3d4
commit ca80944
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/importlib_resources/tests/test_read.py b/importlib_resources/tests/test_read.py
@@ -15,17 +15,26 @@ def execute(self, package, path):
         resources.read(package, path)
 
 
-class ReadTests(unittest.TestCase):
+class ReadTests:
 
     def test_default_encoding(self):
-        result = resources.read(data, 'utf-8.file')
+        result = resources.read(self.data, 'utf-8.file')
         self.assertEqual("Hello, UTF-8 world!\n", result)
 
     def test_encoding(self):
-        result = resources.read(data, 'utf-16.file', encoding='utf-16')
+        result = resources.read(self.data, 'utf-16.file', encoding='utf-16')
         self.assertEqual("Hello, UTF-16 world!\n", result)
 
     def test_errors(self):
         # Raises UnicodeError without the 'errors' argument.
-        result = resources.read(data, 'utf-16.file', encoding='utf-8',
+        result = resources.read(self.data, 'utf-16.file', encoding='utf-8',
                                 errors='ignore')
+
+
+class ReadDiskTests(ReadTests, unittest.TestCase):
+
+    data = data
+
+
+class ReadZipTests(ReadTests, util.ZipSetup, unittest.TestCase):
+    pass