import os
import os.path
from test import cjkencodings_test

dir = 'Lib/test/cjkencodings'

def test_chunkcoding(self):
    tstring_lines = []
    for b in self.tstring:
        lines = b.split(b"\n")
        last = lines.pop()
        assert last == b""
        lines = [line + b"\n" for line in lines]
        tstring_lines.append(lines)
    for native, utf8 in zip(*tstring_lines):
        u = self.decode(native)[0]
        self.assertEqual(u, utf8.decode('utf-8'))
        if self.roundtriptest:
            self.assertEqual(native, self.encode(u)[0])

os.mkdir(dir)
for encoding, data in cjkencodings_test.teststring.items():
    encoded, utf8 = data

    fname = os.path.join(dir, encoding + '.txt')
    with open(fname, 'wb') as f:
        f.write(encoded)

    fname = os.path.join(dir, encoding + '-utf8.txt')
    with open(fname, 'wb') as f:
        f.write(utf8)

os.unlink(cjkencodings_test.__file__)