Skip to content

Commit 981fac0

Browse files
committed
Harmonize pickletester.py imports with RustPython
1 parent d51e253 commit 981fac0

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

Lib/test/pickletester.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,14 @@
2222
_testbuffer = None
2323

2424
from test import support
25+
from test.support import os_helper
2526
from test.support import (
26-
TestFailed, TESTFN, run_with_locale, no_tracing,
27-
_2G, _4G, bigmemtest, reap_threads, forget,
28-
save_restore_warnings_filters
27+
TestFailed, run_with_locale, no_tracing,
28+
_2G, _4G, bigmemtest, reap_threads
2929
)
30+
from test.support.import_helper import forget
31+
from test.support.os_helper import TESTFN
32+
from test.support.warnings_helper import save_restore_warnings_filters
3033

3134
from pickle import bytes_types
3235

@@ -3302,15 +3305,15 @@ def test_dump_closed_file(self):
33023305
f.close()
33033306
self.assertRaises(ValueError, self.dump, 123, f)
33043307
finally:
3305-
support.unlink(TESTFN)
3308+
os_helper.unlink(TESTFN)
33063309

33073310
def test_load_closed_file(self):
33083311
f = open(TESTFN, "wb")
33093312
try:
33103313
f.close()
33113314
self.assertRaises(ValueError, self.dump, 123, f)
33123315
finally:
3313-
support.unlink(TESTFN)
3316+
os_helper.unlink(TESTFN)
33143317

33153318
def test_load_from_and_dump_to_file(self):
33163319
stream = io.BytesIO()
@@ -3341,7 +3344,7 @@ def test_dump_text_file(self):
33413344
self.assertRaises(TypeError, self.dump, 123, f, proto)
33423345
finally:
33433346
f.close()
3344-
support.unlink(TESTFN)
3347+
os_helper.unlink(TESTFN)
33453348

33463349
def test_incomplete_input(self):
33473350
s = io.BytesIO(b"X''.")

Lib/test/test_pickletools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def test__all__(self):
9191
'markobject', 'stackslice', 'OpcodeInfo', 'opcodes',
9292
'code2op',
9393
}
94-
support.check__all__(self, pickletools, blacklist=blacklist)
94+
support.check__all__(self, pickletools, not_exported=blacklist)
9595

9696

9797
def load_tests(loader, tests, pattern):

0 commit comments

Comments
 (0)