| 1 | n/a | # Tests universal newline support for both reading and parsing files. |
|---|
| 2 | 1 | import unittest |
|---|
| 3 | 1 | import os |
|---|
| 4 | 1 | import sys |
|---|
| 5 | 1 | from test import test_support |
|---|
| 6 | n/a | |
|---|
| 7 | 1 | if not hasattr(sys.stdin, 'newlines'): |
|---|
| 8 | 0 | raise unittest.SkipTest, \ |
|---|
| 9 | 0 | "This Python does not have universal newline support" |
|---|
| 10 | n/a | |
|---|
| 11 | 1 | FATX = 'x' * (2**14) |
|---|
| 12 | n/a | |
|---|
| 13 | n/a | DATA_TEMPLATE = [ |
|---|
| 14 | 1 | "line1=1", |
|---|
| 15 | n/a | "line2='this is a very long line designed to go past the magic " + |
|---|
| 16 | n/a | "hundred character limit that is inside fileobject.c and which " + |
|---|
| 17 | 1 | "is meant to speed up the common case, but we also want to test " + |
|---|
| 18 | 1 | "the uncommon case, naturally.'", |
|---|
| 19 | 1 | "def line3():pass", |
|---|
| 20 | 1 | "line4 = '%s'" % FATX, |
|---|
| 21 | n/a | ] |
|---|
| 22 | n/a | |
|---|
| 23 | 1 | DATA_LF = "\n".join(DATA_TEMPLATE) + "\n" |
|---|
| 24 | 1 | DATA_CR = "\r".join(DATA_TEMPLATE) + "\r" |
|---|
| 25 | 1 | DATA_CRLF = "\r\n".join(DATA_TEMPLATE) + "\r\n" |
|---|
| 26 | n/a | |
|---|
| 27 | n/a | # Note that DATA_MIXED also tests the ability to recognize a lone \r |
|---|
| 28 | n/a | # before end-of-file. |
|---|
| 29 | 1 | DATA_MIXED = "\n".join(DATA_TEMPLATE) + "\r" |
|---|
| 30 | 5 | DATA_SPLIT = [x + "\n" for x in DATA_TEMPLATE] |
|---|
| 31 | 1 | del x |
|---|
| 32 | n/a | |
|---|
| 33 | 2 | class TestGenericUnivNewlines(unittest.TestCase): |
|---|
| 34 | n/a | # use a class variable DATA to define the data to write to the file |
|---|
| 35 | n/a | # and a class variable NEWLINE to set the expected newlines value |
|---|
| 36 | 1 | READMODE = 'U' |
|---|
| 37 | 1 | WRITEMODE = 'wb' |
|---|
| 38 | n/a | |
|---|
| 39 | 1 | def setUp(self): |
|---|
| 40 | 26 | with open(test_support.TESTFN, self.WRITEMODE) as fp: |
|---|
| 41 | 26 | fp.write(self.DATA) |
|---|
| 42 | n/a | |
|---|
| 43 | 1 | def tearDown(self): |
|---|
| 44 | 26 | try: |
|---|
| 45 | 26 | os.unlink(test_support.TESTFN) |
|---|
| 46 | 0 | except: |
|---|
| 47 | 0 | pass |
|---|
| 48 | n/a | |
|---|
| 49 | 1 | def test_read(self): |
|---|
| 50 | 5 | with open(test_support.TESTFN, self.READMODE) as fp: |
|---|
| 51 | 5 | data = fp.read() |
|---|
| 52 | 5 | self.assertEqual(data, DATA_LF) |
|---|
| 53 | 5 | self.assertEqual(repr(fp.newlines), repr(self.NEWLINE)) |
|---|
| 54 | n/a | |
|---|
| 55 | 1 | def test_readlines(self): |
|---|
| 56 | 5 | with open(test_support.TESTFN, self.READMODE) as fp: |
|---|
| 57 | 5 | data = fp.readlines() |
|---|
| 58 | 5 | self.assertEqual(data, DATA_SPLIT) |
|---|
| 59 | 5 | self.assertEqual(repr(fp.newlines), repr(self.NEWLINE)) |
|---|
| 60 | n/a | |
|---|
| 61 | 1 | def test_readline(self): |
|---|
| 62 | 5 | with open(test_support.TESTFN, self.READMODE) as fp: |
|---|
| 63 | 5 | data = [] |
|---|
| 64 | 5 | d = fp.readline() |
|---|
| 65 | 25 | while d: |
|---|
| 66 | 20 | data.append(d) |
|---|
| 67 | 20 | d = fp.readline() |
|---|
| 68 | 5 | self.assertEqual(data, DATA_SPLIT) |
|---|
| 69 | 5 | self.assertEqual(repr(fp.newlines), repr(self.NEWLINE)) |
|---|
| 70 | n/a | |
|---|
| 71 | 1 | def test_seek(self): |
|---|
| 72 | 5 | with open(test_support.TESTFN, self.READMODE) as fp: |
|---|
| 73 | 5 | fp.readline() |
|---|
| 74 | 5 | pos = fp.tell() |
|---|
| 75 | 5 | data = fp.readlines() |
|---|
| 76 | 5 | self.assertEqual(data, DATA_SPLIT[1:]) |
|---|
| 77 | 5 | fp.seek(pos) |
|---|
| 78 | 5 | data = fp.readlines() |
|---|
| 79 | 5 | self.assertEqual(data, DATA_SPLIT[1:]) |
|---|
| 80 | n/a | |
|---|
| 81 | 1 | def test_execfile(self): |
|---|
| 82 | 5 | namespace = {} |
|---|
| 83 | 5 | with test_support.check_py3k_warnings(): |
|---|
| 84 | 5 | execfile(test_support.TESTFN, namespace) |
|---|
| 85 | 5 | func = namespace['line3'] |
|---|
| 86 | 5 | self.assertEqual(func.func_code.co_firstlineno, 3) |
|---|
| 87 | 5 | self.assertEqual(namespace['line4'], FATX) |
|---|
| 88 | n/a | |
|---|
| 89 | n/a | |
|---|
| 90 | 2 | class TestNativeNewlines(TestGenericUnivNewlines): |
|---|
| 91 | 1 | NEWLINE = None |
|---|
| 92 | 1 | DATA = DATA_LF |
|---|
| 93 | 1 | READMODE = 'r' |
|---|
| 94 | 1 | WRITEMODE = 'w' |
|---|
| 95 | n/a | |
|---|
| 96 | 2 | class TestCRNewlines(TestGenericUnivNewlines): |
|---|
| 97 | 1 | NEWLINE = '\r' |
|---|
| 98 | 1 | DATA = DATA_CR |
|---|
| 99 | n/a | |
|---|
| 100 | 2 | class TestLFNewlines(TestGenericUnivNewlines): |
|---|
| 101 | 1 | NEWLINE = '\n' |
|---|
| 102 | 1 | DATA = DATA_LF |
|---|
| 103 | n/a | |
|---|
| 104 | 2 | class TestCRLFNewlines(TestGenericUnivNewlines): |
|---|
| 105 | 1 | NEWLINE = '\r\n' |
|---|
| 106 | 1 | DATA = DATA_CRLF |
|---|
| 107 | n/a | |
|---|
| 108 | 1 | def test_tell(self): |
|---|
| 109 | 1 | with open(test_support.TESTFN, self.READMODE) as fp: |
|---|
| 110 | 1 | self.assertEqual(repr(fp.newlines), repr(None)) |
|---|
| 111 | 1 | data = fp.readline() |
|---|
| 112 | 1 | pos = fp.tell() |
|---|
| 113 | 1 | self.assertEqual(repr(fp.newlines), repr(self.NEWLINE)) |
|---|
| 114 | n/a | |
|---|
| 115 | 2 | class TestMixedNewlines(TestGenericUnivNewlines): |
|---|
| 116 | 1 | NEWLINE = ('\r', '\n') |
|---|
| 117 | 1 | DATA = DATA_MIXED |
|---|
| 118 | n/a | |
|---|
| 119 | n/a | |
|---|
| 120 | 1 | def test_main(): |
|---|
| 121 | 1 | test_support.run_unittest( |
|---|
| 122 | 1 | TestNativeNewlines, |
|---|
| 123 | 1 | TestCRNewlines, |
|---|
| 124 | 1 | TestLFNewlines, |
|---|
| 125 | 1 | TestCRLFNewlines, |
|---|
| 126 | 1 | TestMixedNewlines |
|---|
| 127 | n/a | ) |
|---|
| 128 | n/a | |
|---|
| 129 | 1 | if __name__ == '__main__': |
|---|
| 130 | 0 | test_main() |
|---|
