def test_illegal_index():
d = SortedDict()
with pytest.raises(IndexError):
d.index(0)
with pytest.raises(TypeError):
d.index('a')
with pytest.raises(TypeError):
d.index()
def test_index():
s = SortedDict(ordering='DESC')
s[1] = "a"
s[3] = "b"
s[2] = "c"
assert s.index(0) == (3, "b")
assert s.index(1) == (2, "c")
assert s.index(2) == (1, "a")
assert s.index(-1) == (1, 'a')
assert s.index(-2) == (2, 'c')
assert s.index(-3) == (3, 'b')
with pytest.raises(IndexError):
assert s.index(3) == (2, "c")
with pytest.raises(IndexError):
assert s.index(4) == (2, "c")