mirror of
https://github.com/python/cpython.git
synced 2025-11-02 19:12:55 +00:00
Move assertBytesEqual to base test class, improve it, and hook into assertEqual
This commit is contained in:
parent
eb50e51a0f
commit
a46ed1186f
2 changed files with 12 additions and 6 deletions
|
|
@ -3161,12 +3161,7 @@ class BaseTestBytesGeneratorIdempotent:
|
|||
b = BytesIO()
|
||||
g = email.generator.BytesGenerator(b, maxheaderlen=0)
|
||||
g.flatten(msg, unixfrom=unixfrom, linesep=self.linesep)
|
||||
self.assertByteStringsEqual(data, b.getvalue())
|
||||
|
||||
def assertByteStringsEqual(self, str1, str2):
|
||||
# Not using self.blinesep here is intentional. This way the output
|
||||
# is more useful when the failure results in mixed line endings.
|
||||
self.assertListEqual(str1.split(b'\n'), str2.split(b'\n'))
|
||||
self.assertEqual(data, b.getvalue())
|
||||
|
||||
|
||||
class TestBytesGeneratorIdempotentNL(BaseTestBytesGeneratorIdempotent,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue