37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from benedict.utils import keylist_util
|
|
|
|
import unittest
|
|
|
|
|
|
class keylist_util_test_case(unittest.TestCase):
|
|
|
|
def test_set_item_with_indexes(self):
|
|
d = {}
|
|
|
|
keylist_util.set_item(d, 'a', None)
|
|
self.assertEqual(d, {'a':None})
|
|
|
|
keylist_util.set_item(d, ['a', 'b', 'c'], 0)
|
|
self.assertEqual(d, {'a':{'b':{'c':0}}})
|
|
|
|
keylist_util.set_item(d, ['a', 'b', 'd'], 1)
|
|
self.assertEqual(d, {'a':{'b':{'c':0,'d':1}}})
|
|
# '[1]'
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 0], 1)
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 1], 2)
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 2], 3)
|
|
self.assertEqual(d, { 'a':{ 'b':{ 'c':0, 'd':1, 'e':[1, 2, 3] }}})
|
|
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 0], 4)
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 1], 5)
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 2], 6)
|
|
# keylist_util.set_item(d, ['a', 'b', 'e', 3], 7)
|
|
# keylist_util.set_item(d, ['a', 'b', 'e', 4], 8)
|
|
keylist_util.set_item(d, ['a', 'b', 'e', 5], 9)
|
|
self.assertEqual(d, { 'a':{ 'b':{ 'c':0, 'd':1, 'e':[4, 5, 6, None, None, 9] }}})
|
|
keylist_util.set_item(d, ['a', 'b', 'e', -11], 10)
|
|
self.assertEqual(d, { 'a':{ 'b':{ 'c':0, 'd':1, 'e':[10, 4, 5, 6, None, None, 9] }}})
|
|
|