115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
# Tests universal newline support for both reading and parsing files.
|
|
import unittest
|
|
import test_support
|
|
import os
|
|
import sys
|
|
|
|
DATA_TEMPLATE=[
|
|
"line1=1",
|
|
"line2='this is a very long line designed to go past the magic " +
|
|
"hundred character limit that is inside fileobject.c and which " +
|
|
"is meant to speed up the common case, but we also want to test " +
|
|
"the uncommon case, naturally.'",
|
|
"def line3():pass"
|
|
]
|
|
DATA_LF = "\n".join(DATA_TEMPLATE) + "\n"
|
|
DATA_CR = "\r".join(DATA_TEMPLATE) + "\r"
|
|
DATA_CRLF = "\r\n".join(DATA_TEMPLATE) + "\r\n"
|
|
# Note that DATA_MIXED also tests the ability to recognize a lone \r
|
|
# before end-of-file.
|
|
DATA_MIXED = "\n".join(DATA_TEMPLATE) + "\r"
|
|
DATA_SPLIT = map(lambda x: x+"\n", DATA_TEMPLATE)
|
|
|
|
if not hasattr(sys.stdin, 'newlines'):
|
|
raise test_support.TestSkipped, \
|
|
"This Python does not have universal newline support"
|
|
|
|
class TestGenericUnivNewlines(unittest.TestCase):
|
|
# use a class variable DATA to define the data to write to the file
|
|
# and a class variable NEWLINE to set the expected newlines value
|
|
READMODE = 'U'
|
|
WRITEMODE = 'wb'
|
|
|
|
def setUp(self):
|
|
fp = open(test_support.TESTFN, self.WRITEMODE)
|
|
fp.write(self.DATA)
|
|
fp.close()
|
|
|
|
def tearDown(self):
|
|
try:
|
|
os.unlink(test_support.TESTFN)
|
|
except:
|
|
pass
|
|
|
|
def test_read(self):
|
|
fp = open(test_support.TESTFN, self.READMODE)
|
|
data = fp.read()
|
|
self.assertEqual(data, DATA_LF)
|
|
self.assertEqual(`fp.newlines`, `self.NEWLINE`)
|
|
|
|
def test_readlines(self):
|
|
fp = open(test_support.TESTFN, self.READMODE)
|
|
data = fp.readlines()
|
|
self.assertEqual(data, DATA_SPLIT)
|
|
self.assertEqual(`fp.newlines`, `self.NEWLINE`)
|
|
|
|
def test_readline(self):
|
|
fp = open(test_support.TESTFN, self.READMODE)
|
|
data = []
|
|
d = fp.readline()
|
|
while d:
|
|
data.append(d)
|
|
d = fp.readline()
|
|
self.assertEqual(data, DATA_SPLIT)
|
|
self.assertEqual(`fp.newlines`, `self.NEWLINE`)
|
|
|
|
def test_seek(self):
|
|
fp = open(test_support.TESTFN, self.READMODE)
|
|
fp.readline()
|
|
pos = fp.tell()
|
|
data = fp.readlines()
|
|
self.assertEqual(data, DATA_SPLIT[1:])
|
|
fp.seek(pos)
|
|
data = fp.readlines()
|
|
self.assertEqual(data, DATA_SPLIT[1:])
|
|
|
|
def test_execfile(self):
|
|
dict = {}
|
|
execfile(test_support.TESTFN, dict)
|
|
func = dict['line3']
|
|
self.assertEqual(func.func_code.co_firstlineno, 3)
|
|
|
|
|
|
class TestNativeNewlines(TestGenericUnivNewlines):
|
|
NEWLINE=None
|
|
DATA=DATA_LF
|
|
READMODE='r'
|
|
WRITEMODE='w'
|
|
|
|
class TestCRNewlines(TestGenericUnivNewlines):
|
|
NEWLINE='\r'
|
|
DATA=DATA_CR
|
|
|
|
class TestLFNewlines(TestGenericUnivNewlines):
|
|
NEWLINE='\n'
|
|
DATA=DATA_LF
|
|
|
|
class TestCRLFNewlines(TestGenericUnivNewlines):
|
|
NEWLINE='\r\n'
|
|
DATA=DATA_CRLF
|
|
|
|
class TestMixedNewlines(TestGenericUnivNewlines):
|
|
NEWLINE=('\r', '\n')
|
|
DATA=DATA_MIXED
|
|
|
|
|
|
def test_main():
|
|
test_support.run_unittest(TestNativeNewlines)
|
|
test_support.run_unittest(TestCRNewlines)
|
|
test_support.run_unittest(TestLFNewlines)
|
|
test_support.run_unittest(TestCRLFNewlines)
|
|
test_support.run_unittest(TestMixedNewlines)
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|