mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-01 22:47:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Fix expected memory consumption for test_translate

Browse source
This commit is contained in:
Antoine Pitrou 2011-10-06 22:41:08 +02:00
parent b6080b6967
commit 399df55496
1 changed files with 25 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

33
Lib/test/test_bigmem.py
View file

@ -446,14 +446,7 @@ class BaseStrTest:
def test_translate(self, size): def test_translate(self, size):
_ = self.from_latin1 _ = self.from_latin1
SUBSTR = _('aZz.z.Aaz.') SUBSTR = _('aZz.z.Aaz.')
if isinstance(SUBSTR, str): trans = bytes.maketrans(b'.aZ', b'-!$')
trans = {
ord(_('.')): _('-'),
ord(_('a')): _('!'),
ord(_('Z')): _('$'),
}
else:
trans = bytes.maketrans(b'.aZ', b'-!$')
sublen = len(SUBSTR) sublen = len(SUBSTR)
repeats = size // sublen + 2 repeats = size // sublen + 2
s = SUBSTR * repeats s = SUBSTR * repeats
@ -735,6 +728,30 @@ class StrTest(unittest.TestCase, BaseStrTest):
finally: finally:
r = s = None r = s = None
# The original test_translate is overriden here, so as to get the
# correct size estimate: str.translate() uses an intermediate Py_UCS4
# representation.
@bigmemtest(size=_2G, memuse=ascii_char_size * 2 + ucs4_char_size)
def test_translate(self, size):
_ = self.from_latin1
SUBSTR = _('aZz.z.Aaz.')
trans = {
ord(_('.')): _('-'),
ord(_('a')): _('!'),
ord(_('Z')): _('$'),
}
sublen = len(SUBSTR)
repeats = size // sublen + 2
s = SUBSTR * repeats
s = s.translate(trans)
self.assertEqual(len(s), repeats * sublen)
self.assertEqual(s[:sublen], SUBSTR.translate(trans))
self.assertEqual(s[-sublen:], SUBSTR.translate(trans))
self.assertEqual(s.count(_('.')), 0)
self.assertEqual(s.count(_('!')), repeats * 2)
self.assertEqual(s.count(_('z')), repeats * 3)
class BytesTest(unittest.TestCase, BaseStrTest): class BytesTest(unittest.TestCase, BaseStrTest):