159 lines
5.8 KiB
Python
159 lines
5.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import unittest
|
|
from nose import tools
|
|
|
|
from kitchen.pycompat24.sets import add_builtin_set
|
|
add_builtin_set()
|
|
|
|
from kitchen import collections
|
|
|
|
def test_strict_dict_get_set():
|
|
'''Test getting and setting items in StrictDict'''
|
|
d = collections.StrictDict()
|
|
d['a'] = 1
|
|
d[b'a'] = 2
|
|
tools.assert_not_equal(d['a'], d[b'a'])
|
|
tools.eq_(len(d), 2)
|
|
|
|
d['\xf1'] = 1
|
|
d[b'\xf1'] = 2
|
|
d['\xf1'.encode('utf-8')] = 3
|
|
tools.eq_(d['\xf1'], 1)
|
|
tools.eq_(d[b'\xf1'], 2)
|
|
tools.eq_(d['\xf1'.encode('utf-8')], 3)
|
|
tools.eq_(len(d), 5)
|
|
|
|
class TestStrictDict(unittest.TestCase):
|
|
def setUp(self):
|
|
self.d = collections.StrictDict()
|
|
self.d['a'] = 1
|
|
self.d[b'a'] = 2
|
|
self.d['\xf1'] = 1
|
|
self.d[b'\xf1'] = 2
|
|
self.d['\xf1'.encode('utf-8')] = 3
|
|
self.keys = ['a', b'a', '\xf1', b'\xf1', '\xf1'.encode('utf-8')]
|
|
|
|
def tearDown(self):
|
|
del(self.d)
|
|
|
|
def _compare_lists(self, list1, list2, debug=False):
|
|
'''We have a mixture of bytes and unicode so we have to compare these
|
|
lists manually and inefficiently
|
|
'''
|
|
def _compare_lists_helper(compare_to, dupes, idx, length):
|
|
if i not in compare_to:
|
|
return False
|
|
for n in range(1, length + 1):
|
|
if i not in dupes[n][idx]:
|
|
dupes[n][idx].add(i)
|
|
return True
|
|
if len(list1) != len(list2):
|
|
return False
|
|
|
|
list1_dupes = dict([(i, (set(), set(), set())) for i in range(1, len(list1)+1)])
|
|
list2_dupes = dict([(i, (set(), set(), set())) for i in range(1, len(list1)+1)])
|
|
|
|
list1_u = [l for l in list1 if isinstance(l, str)]
|
|
list1_b = [l for l in list1 if isinstance(l, bytes)]
|
|
list1_o = [l for l in list1 if not (isinstance(l, (str, bytes)))]
|
|
|
|
list2_u = [l for l in list2 if isinstance(l, str)]
|
|
list2_b = [l for l in list2 if isinstance(l, bytes)]
|
|
list2_o = [l for l in list2 if not (isinstance(l, (str, bytes)))]
|
|
|
|
for i in list1:
|
|
if isinstance(i, str):
|
|
if not _compare_lists_helper(list2_u, list1_dupes, 0, len(list1)):
|
|
return False
|
|
elif isinstance(i, bytes):
|
|
if not _compare_lists_helper(list2_b, list1_dupes, 1, len(list1)):
|
|
return False
|
|
else:
|
|
if not _compare_lists_helper(list2_o, list1_dupes, 2, len(list1)):
|
|
return False
|
|
|
|
if list1_dupes[2][0] or list1_dupes[2][1] or list1_dupes[2][2]:
|
|
for i in list2:
|
|
if isinstance(i, str):
|
|
if not _compare_lists_helper(list1_u, list2_dupes, 0, len(list1)):
|
|
return False
|
|
elif isinstance(i, bytes):
|
|
if not _compare_lists_helper(list1_b, list2_dupes, 1, len(list1)):
|
|
return False
|
|
else:
|
|
if not _compare_lists_helper(list1_o, list2_dupes, 2, len(list1)):
|
|
return False
|
|
|
|
for i in range(2, len(list1)+1):
|
|
for n in list1_dupes[i]:
|
|
if n not in list2_dupes[i]:
|
|
return False
|
|
|
|
return True
|
|
|
|
def test__compare_list(self):
|
|
'''*sigh* this test support function is so complex we need to test it'''
|
|
tools.ok_(self._compare_lists(['a', 'b', 'c'], ['c', 'a', 'b']))
|
|
tools.ok_(not self._compare_lists(['b', 'c'], ['c', 'a', 'b']))
|
|
tools.ok_(not self._compare_lists([b'a', 'b'], ['a', 'b']))
|
|
tools.ok_(not self._compare_lists(['a', b'b'], [b'a', 'b']))
|
|
tools.ok_(self._compare_lists(['a', 'b', 1], ['a', 1, 'b']))
|
|
tools.ok_(self._compare_lists([b'a', b'b'], [b'a', b'b']))
|
|
tools.ok_(self._compare_lists([b'a', 'b'], [b'a', 'b']))
|
|
tools.ok_(not self._compare_lists([b'a', 'b'], [b'a', b'b']))
|
|
tools.ok_(self._compare_lists([b'a', 'b', 'b', 'c', b'a'], [b'a', b'a', 'b', 'c', 'b']))
|
|
tools.ok_(not self._compare_lists([b'a', 'b', 'b', 'c', 'a'], [b'a', b'a', 'b', 'c', 'b']))
|
|
tools.ok_(not self._compare_lists([b'a', 'b', 'b', 'c', b'a'], [b'a', 'b', 'b', 'c', 'b']))
|
|
|
|
def test_strict_dict_len(self):
|
|
'''StrictDict len'''
|
|
tools.eq_(len(self.d), 5)
|
|
|
|
def test_strict_dict_del(self):
|
|
'''StrictDict del'''
|
|
tools.eq_(len(self.d), 5)
|
|
del(self.d['\xf1'])
|
|
tools.assert_raises(KeyError, self.d.__getitem__, '\xf1')
|
|
tools.eq_(len(self.d), 4)
|
|
|
|
def test_strict_dict_iter(self):
|
|
'''StrictDict iteration'''
|
|
keys = []
|
|
for k in self.d:
|
|
keys.append(k)
|
|
tools.ok_(self._compare_lists(keys, self.keys),
|
|
msg='keys != self.key: %s != %s' % (keys, self.keys))
|
|
|
|
keys = []
|
|
for k in self.d.keys():
|
|
keys.append(k)
|
|
tools.ok_(self._compare_lists(keys, self.keys),
|
|
msg='keys != self.key: %s != %s' % (keys, self.keys))
|
|
|
|
keys = [k for k in self.d]
|
|
tools.ok_(self._compare_lists(keys, self.keys),
|
|
msg='keys != self.key: %s != %s' % (keys, self.keys))
|
|
|
|
keys = []
|
|
for k in list(self.d.keys()):
|
|
keys.append(k)
|
|
tools.ok_(self._compare_lists(keys, self.keys),
|
|
msg='keys != self.key: %s != %s' % (keys, self.keys))
|
|
|
|
def test_strict_dict_contains(self):
|
|
'''StrictDict contains function'''
|
|
tools.ok_(b'b' not in self.d)
|
|
tools.ok_('b' not in self.d)
|
|
tools.ok_(b'\xf1' in self.d)
|
|
tools.ok_('\xf1' in self.d)
|
|
tools.ok_(b'a' in self.d)
|
|
tools.ok_('a' in self.d)
|
|
|
|
del(self.d['\xf1'])
|
|
tools.ok_('\xf1' not in self.d)
|
|
tools.ok_(b'\xf1' in self.d)
|
|
|
|
del(self.d[b'a'])
|
|
tools.ok_('a' in self.d)
|
|
tools.ok_(b'a' not in self.d)
|