mirror of https://github.com/python/cpython
118 lines
3.7 KiB
Python
118 lines
3.7 KiB
Python
from test.support import open_urlresource
|
|
import unittest
|
|
|
|
from http.client import HTTPException
|
|
import sys
|
|
from unicodedata import normalize, is_normalized, unidata_version
|
|
|
|
TESTDATAFILE = "NormalizationTest.txt"
|
|
TESTDATAURL = "http://www.pythontest.net/unicode/" + unidata_version + "/" + TESTDATAFILE
|
|
|
|
def check_version(testfile):
|
|
hdr = testfile.readline()
|
|
return unidata_version in hdr
|
|
|
|
class RangeError(Exception):
|
|
pass
|
|
|
|
def NFC(str):
|
|
return normalize("NFC", str)
|
|
|
|
def NFKC(str):
|
|
return normalize("NFKC", str)
|
|
|
|
def NFD(str):
|
|
return normalize("NFD", str)
|
|
|
|
def NFKD(str):
|
|
return normalize("NFKD", str)
|
|
|
|
def unistr(data):
|
|
data = [int(x, 16) for x in data.split(" ")]
|
|
for x in data:
|
|
if x > sys.maxunicode:
|
|
raise RangeError
|
|
return "".join([chr(x) for x in data])
|
|
|
|
class NormalizationTest(unittest.TestCase):
|
|
def test_main(self):
|
|
# Hit the exception early
|
|
try:
|
|
testdata = open_urlresource(TESTDATAURL, encoding="utf-8",
|
|
check=check_version)
|
|
except PermissionError:
|
|
self.skipTest(f"Permission error when downloading {TESTDATAURL} "
|
|
f"into the test data directory")
|
|
except (OSError, HTTPException):
|
|
self.fail(f"Could not retrieve {TESTDATAURL}")
|
|
|
|
with testdata:
|
|
self.run_normalization_tests(testdata)
|
|
|
|
def run_normalization_tests(self, testdata):
|
|
part = None
|
|
part1_data = {}
|
|
|
|
for line in testdata:
|
|
if '#' in line:
|
|
line = line.split('#')[0]
|
|
line = line.strip()
|
|
if not line:
|
|
continue
|
|
if line.startswith("@Part"):
|
|
part = line.split()[0]
|
|
continue
|
|
try:
|
|
c1,c2,c3,c4,c5 = [unistr(x) for x in line.split(';')[:-1]]
|
|
except RangeError:
|
|
# Skip unsupported characters;
|
|
# try at least adding c1 if we are in part1
|
|
if part == "@Part1":
|
|
try:
|
|
c1 = unistr(line.split(';')[0])
|
|
except RangeError:
|
|
pass
|
|
else:
|
|
part1_data[c1] = 1
|
|
continue
|
|
|
|
# Perform tests
|
|
self.assertTrue(c2 == NFC(c1) == NFC(c2) == NFC(c3), line)
|
|
self.assertTrue(c4 == NFC(c4) == NFC(c5), line)
|
|
self.assertTrue(c3 == NFD(c1) == NFD(c2) == NFD(c3), line)
|
|
self.assertTrue(c5 == NFD(c4) == NFD(c5), line)
|
|
self.assertTrue(c4 == NFKC(c1) == NFKC(c2) == \
|
|
NFKC(c3) == NFKC(c4) == NFKC(c5),
|
|
line)
|
|
self.assertTrue(c5 == NFKD(c1) == NFKD(c2) == \
|
|
NFKD(c3) == NFKD(c4) == NFKD(c5),
|
|
line)
|
|
|
|
self.assertTrue(is_normalized("NFC", c2))
|
|
self.assertTrue(is_normalized("NFC", c4))
|
|
|
|
self.assertTrue(is_normalized("NFD", c3))
|
|
self.assertTrue(is_normalized("NFD", c5))
|
|
|
|
self.assertTrue(is_normalized("NFKC", c4))
|
|
self.assertTrue(is_normalized("NFKD", c5))
|
|
|
|
# Record part 1 data
|
|
if part == "@Part1":
|
|
part1_data[c1] = 1
|
|
|
|
# Perform tests for all other data
|
|
for c in range(sys.maxunicode+1):
|
|
X = chr(c)
|
|
if X in part1_data:
|
|
continue
|
|
self.assertTrue(X == NFC(X) == NFD(X) == NFKC(X) == NFKD(X), c)
|
|
|
|
def test_bug_834676(self):
|
|
# Check for bug 834676
|
|
normalize('NFC', '\ud55c\uae00')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|