def test_keys_reference_counting():
d = SortedDict()
d[1] = 'a'
assert d.keys() == (1, )
d[2] = 'b'
assert d.keys() == (1, 2)
def test_init_from_dict():
with pytest.raises(TypeError):
asc = SortedDict("a", ordering='ASC')
with pytest.raises(TypeError):
asc = SortedDict({}, {}, ordering='ASC')
asc = SortedDict({
4: 'a',
1: 'c',
3: 'f',
6: 'j',
9: 'z',
2: 'p'
},
ordering='ASC')
assert asc.to_dict() == {1: 'c', 2: 'p', 3: 'f', 4: 'a', 6: 'j', 9: 'z'}
assert list(asc.keys()) == [1, 2, 3, 4, 6, 9]
desc = SortedDict({
4: 'a',
1: 'c',
3: 'f',
6: 'j',
9: 'z',
2: 'p'
},
ordering='DESC')
assert desc.to_dict() == {1: 'c', 2: 'p', 3: 'f', 4: 'a', 6: 'j', 9: 'z'}
assert list(desc.keys()) == [9, 6, 4, 3, 2, 1]
def test_to_dict():
random.seed()
values = []
asc = SortedDict(ordering='ASC')
desc = SortedDict(ordering='DESC')
for _ in range(2000):
values.append(random.uniform(0.0, 100000.0))
values = set(values)
for v in values:
asc[v] = str(v)
desc[v] = str(v)
d = asc.to_dict()
assert list(d.keys()) == list(asc.keys())
assert sorted(d.keys()) == list(d.keys())
previous = None
for key, _ in d.items():
assert d[key] == asc[key]
if previous:
d[key] > previous
previous = d[key]
d = desc.to_dict()
assert list(d.keys()) == list(desc.keys())
assert list(reversed(sorted(d.keys()))) == list(d.keys())
previous = None
for key, val in d.items():
assert d[key] == desc[key]
if previous:
d[key] < previous
previous = d[key]
def test_depth_auto_truncate():
d = SortedDict({i: i for i in range(100)}, truncate=True, max_depth=10)
del d[5]
assert d.keys() == (0, 1, 2, 3, 4, 6, 7, 8, 9)
d[1.1] = 0
d[1.2] = 0
d[1.3] = 0
assert d.keys() == (0, 1, 1.1, 1.2, 1.3, 2, 3, 4, 6, 7)
def test_descending():
s = SortedDict(ordering='DESC')
s[1] = "a"
s[3] = "b"
s[2] = "c"
assert s.keys() == (3, 2, 1)
def test_ascending():
s = SortedDict(ordering='ASC')
s[3] = "a"
s[2] = "b"
s[1] = "c"
assert s.keys() == (1, 2, 3)
def test_default_ordering():
# default ordering is ascending
d = SortedDict()
d[3] = 'a'
d[2] = 'b'
d[1] = 'c'
assert list(d.keys()) == [1, 2, 3]
def test_del():
d = SortedDict()
d[3] = 'a'
d[2] = 'b'
d[1] = 'c'
del d[2]
assert d.keys() == (1, 3)
def test_decimal():
s = SortedDict(ordering='DESC')
s[Decimal('1.2')] = "a"
s[Decimal('1.5')] = "b"
s[Decimal('1.6')] = "c"
s[Decimal('1.7')] = "d"
assert len(s) == 4
assert s.keys() == (Decimal('1.7'), Decimal('1.6'), Decimal('1.5'),
Decimal('1.2'))
def test_depth_truncated():
d = SortedDict({i: i for i in range(100)}, max_depth=10)
d.truncate()
del d[5]
assert d.keys() == (0, 1, 2, 3, 4, 6, 7, 8, 9)
def test_depth():
d = SortedDict({i: i for i in range(100)}, max_depth=10)
assert d.keys() == (0, 1, 2, 3, 4, 5, 6, 7, 8, 9)
assert len(d) == 10
def test_empty_keys():
d = SortedDict()
assert d.keys() == ()
