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>', )