def test_single_encode_generators(self):
def gen1():
yield b'hurr<>durr'
def gen2():
yield 12
yield '<abc>'
yield gen1()
self.assertTrue(isinstance(
encode.single_encode(gen2(), True, True, allow_generators=False, stringify=True),
bytes
))
self.assertSequenceEqual(
list(encode.single_encode(gen2(), True, True, allow_generators=True, stringify=True)),
[b'12', b'<abc>', b'hurr<>durr'],
)
def test_single_encode(self):
self.assertEqual(
encode.single_encode(None),
None,
)
self.assertEqual(
encode.single_encode(12),
12,
)
self.assertEqual(
encode.single_encode('abc'),
b'abc',
)
self.assertEqual(
encode.single_encode('abc', False),
'abc',
)
self.assertEqual(
encode.single_encode('<abc>', False, True),
'<abc>',
)
self.assertEqual(
encode.single_encode('<abc>', True, True),
b'<abc>',
)
self.assertEqual(
encode.single_encode('zażółć', encoding='iso-8859-2'),
'zażółć'.encode('iso-8859-2'),
)
self.assertEqual(
encode.single_encode('zażółć', encoding='utf8'),
'zażółć'.encode('utf8'),
)
self.assertEqual(
encode.single_encode(None, stringify=True),
b'None',
)
self.assertEqual(
encode.single_encode(12, stringify=True),
b'12',
)
self.assertEqual(
encode.single_encode(b'<abc>', True, True, stringify=True),
b'<abc>',
)
