Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Include some tests for non_numeric_sort_collator

Browse Source
This commit is contained in:
Charles Haley 2021-10-24 17:08:15 +01:00
parent 06b598eb9d
commit 82e4b04e2f
1 changed files with 29 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/calibre/utils/icu_test.py
View File

@ -25,6 +25,7 @@ def make_collation_func(name, locale, numeric=True, maker=icu.make_sort_key_func
class TestICU(unittest.TestCase): class TestICU(unittest.TestCase):
ae = unittest.TestCase.assertEqual ae = unittest.TestCase.assertEqual
ane= unittest.TestCase.assertNotEqual
def setUp(self): def setUp(self):
icu.change_locale('en') icu.change_locale('en')
@ -115,15 +116,38 @@ class TestICU(unittest.TestCase):
def test_collation_order(self): def test_collation_order(self):
'Testing collation ordering' 'Testing collation ordering'
from calibre.utils.icu import collation_order
for group in [ for group in [
('Šaa', 'Smith', 'Solženicyn', 'Štepánek'), (self.ae, ('Šaa', 'Smith', 'Solženicyn', 'Štepánek')),
('01', '1'), (self.ae, ('11', '011')),
(self.ane, ('2', '1')),
(self.ae, ('100 Smith', '0100 Smith')),
]: ]:
last = None last = None
for x in group: assert_func = group[0]
order, length = icu.numeric_collator().collation_order(x) for x in group[1]:
order, _ = icu.numeric_collator().collation_order(x)
if last is not None: if last is not None:
self.ae(last, order, 'Order for %s not correct: %s != %s' % (x, last, order)) assert_func(last, order, 'Order for %s not correct: %s != %s' % (x, last, order))
last = order
self.ae(dict(icu.partition_by_first_letter(['A1', '', 'a1', '\U0001f431', '\U0001f431x'])),
{' ':[''], 'A':['A1', 'a1'], '\U0001f431':['\U0001f431', '\U0001f431x']})
def test_collation_order_for_partitioning(self):
'Testing collation ordering for partitioning'
for group in [
(self.ae, ('Smith', 'Šaa', 'Solženicyn', 'Štepánek')),
(self.ane, ('11', '011')),
(self.ae, ('102 Smith', '100 Smith')),
(self.ane, ('100 Smith', '0100 Smith')),
]:
last = None
assert_func = group[0]
for x in group[1]:
order, _ = icu.non_numeric_sort_collator().collation_order(x)
if last is not None:
assert_func(last, order, 'Order for %s not correct: %s != %s' % (x, last, order))
last = order last = order
self.ae(dict(icu.partition_by_first_letter(['A1', '', 'a1', '\U0001f431', '\U0001f431x'])), self.ae(dict(icu.partition_by_first_letter(['A1', '', 'a1', '\U0001f431', '\U0001f431x'])),