Skip to content

Commit 2b2ecc4

Browse files
committed
Fix Ascii85 test from mainline
1 parent 8adaf2c commit 2b2ecc4

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

Lib/test/test_base64.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -664,17 +664,20 @@ def test_a85encode_wrapcol(self):
664664
b'<~\nGB\n\\6\n`E\n-Z\nP=\nDf\n.1\nGE\nb>\n~>')
665665
eq(base64.a85encode(b, wrapcol=sys.maxsize), b'GB\\6`E-ZP=Df.1GEb>')
666666
if check_impl_detail():
667-
eq(base64.a85encode(b, wrapcol=2**1000), b'GB\\6`E-ZP=Df.1GEb>')
668-
eq(base64.a85encode(b, wrapcol=-7),
669-
b'G\nB\n\\\n6\n`\nE\n-\nZ\nP\n=\nD\nf\n.\n1\nG\nE\nb\n>')
670-
eq(base64.a85encode(b, wrapcol=-7, adobe=True),
671-
b'<~\nGB\n\\6\n`E\n-Z\nP=\nDf\n.1\nGE\nb>\n~>')
667+
eq(base64.a85encode(b, wrapcol=sys.maxsize*2),
668+
b'GB\\6`E-ZP=Df.1GEb>')
669+
with self.assertRaises(OverflowError):
670+
base64.a85encode(b, wrapcol=2**1000)
671+
with self.assertRaises(ValueError):
672+
base64.a85encode(b, wrapcol=-7)
673+
with self.assertRaises(ValueError):
674+
base64.a85encode(b, wrapcol=-7, adobe=True)
672675
with self.assertRaises(TypeError):
673676
base64.a85encode(b, wrapcol=7.0)
674677
with self.assertRaises(TypeError):
675678
base64.a85encode(b, wrapcol='7')
676-
if check_impl_detail():
677-
eq(base64.a85encode(b, wrapcol=None), b'GB\\6`E-ZP=Df.1GEb>')
679+
with self.assertRaises(TypeError):
680+
base64.a85encode(b, wrapcol=None)
678681
eq(base64.a85encode(b'', wrapcol=0), b'')
679682
eq(base64.a85encode(b'', wrapcol=7), b'')
680683
eq(base64.a85encode(b'', wrapcol=1, adobe=True), b'<~\n~>')

0 commit comments

Comments
 (0)