applied suggestion by erlend-aasland

python · erlend-aasland · May 7, 2023 · May 7, 2023 · May 7, 2023 · May 7, 2023
commit 8255467fac91c691367f1b38a1a89a28f68dd5bc
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py
@@ -7,11 +7,10 @@
 from io import StringIO
 from tempfile import TemporaryFile
 import csv
-import _csv
 import gc
 import pickle
 from test import support
-from test.support import warnings_helper, check_disallow_instantiation
+from test.support import warnings_helper, import_helper, check_disallow_instantiation
 from itertools import permutations
 from textwrap import dedent
 from collections import OrderedDict
@@ -1431,11 +1430,12 @@ def test_subclassable(self):
        # issue 44089
        class Foo(csv.Error): ...

-    def test_reader_disallow_instantiation(self):
-        check_disallow_instantiation(self, _csv.Reader)
-
-    def test_writer_disallow_instantiation(self):
-        check_disallow_instantiation(self, _csv.Writer)
+    @support.cpython_only
+    def test_disallow_instantiation(self):
+        _csv = import_helper.import_module("_csv")
+        for tp in _csv.Reader, _csv.Writer:
+            with self.subTest(tp=tp):
+                check_disallow_instantiation(self, tp)

 if __name__ == '__main__':
    unittest.main()