Skip to content

Commit 0cec45d

Browse files
committed
with open(), remove test_array from TSAN tests
1 parent 684b953 commit 0cec45d

File tree

2 files changed

+19
-29
lines changed

2 files changed

+19
-29
lines changed

Lib/test/libregrtest/tsan.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
# chosen because they use threads and run in a reasonable amount of time.
33

44
TSAN_TESTS = [
5-
'test_array',
65
'test_asyncio',
76
'test_capi',
87
'test_code',

Lib/test/test_array.py

Lines changed: 19 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -475,22 +475,18 @@ def test_tofromfile(self):
475475
self.assertRaises(TypeError, a.tofile)
476476
with os_helper.temp_dir() as temp_dir:
477477
temp_path = os.path.join(temp_dir, os_helper.TESTFN)
478-
f = open(temp_path, 'wb')
479478
try:
480-
a.tofile(f)
481-
f.close()
479+
with open(temp_path, 'wb') as f:
480+
a.tofile(f)
482481
b = array.array(self.typecode)
483-
f = open(temp_path, 'rb')
484-
self.assertRaises(TypeError, b.fromfile)
485-
b.fromfile(f, len(self.example))
486-
self.assertEqual(b, array.array(self.typecode, self.example))
487-
self.assertNotEqual(a, b)
488-
self.assertRaises(EOFError, b.fromfile, f, len(self.example)+1)
489-
self.assertEqual(a, b)
490-
f.close()
482+
with open(temp_path, 'rb') as f:
483+
self.assertRaises(TypeError, b.fromfile)
484+
b.fromfile(f, len(self.example))
485+
self.assertEqual(b, array.array(self.typecode, self.example))
486+
self.assertNotEqual(a, b)
487+
self.assertRaises(EOFError, b.fromfile, f, len(self.example)+1)
488+
self.assertEqual(a, b)
491489
finally:
492-
if not f.closed:
493-
f.close()
494490
os_helper.unlink(temp_path)
495491

496492
def test_fromfile_ioerror(self):
@@ -499,32 +495,27 @@ def test_fromfile_ioerror(self):
499495
a = array.array(self.typecode)
500496
with os_helper.temp_dir() as temp_dir:
501497
temp_path = os.path.join(temp_dir, os_helper.TESTFN)
502-
f = open(temp_path, 'wb')
503498
try:
504-
self.assertRaises(OSError, a.fromfile, f, len(self.example))
499+
with open(temp_path, 'wb') as f:
500+
self.assertRaises(OSError, a.fromfile, f, len(self.example))
505501
finally:
506-
f.close()
507502
os_helper.unlink(temp_path)
508503

509504
def test_filewrite(self):
510505
a = array.array(self.typecode, 2*self.example)
511506
with os_helper.temp_dir() as temp_dir:
512507
temp_path = os.path.join(temp_dir, os_helper.TESTFN)
513-
f = open(temp_path, 'wb')
514508
try:
515-
f.write(a)
516-
f.close()
509+
with open(temp_path, 'wb') as f:
510+
f.write(a)
517511
b = array.array(self.typecode)
518-
f = open(temp_path, 'rb')
519-
b.fromfile(f, len(self.example))
520-
self.assertEqual(b, array.array(self.typecode, self.example))
521-
self.assertNotEqual(a, b)
522-
b.fromfile(f, len(self.example))
523-
self.assertEqual(a, b)
524-
f.close()
512+
with open(temp_path, 'rb') as f:
513+
b.fromfile(f, len(self.example))
514+
self.assertEqual(b, array.array(self.typecode, self.example))
515+
self.assertNotEqual(a, b)
516+
b.fromfile(f, len(self.example))
517+
self.assertEqual(a, b)
525518
finally:
526-
if not f.closed:
527-
f.close()
528519
os_helper.unlink(temp_path)
529520

530521
def test_tofromlist(self):

0 commit comments

Comments
 (0)