52 lines
996 B
Python
52 lines
996 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from benedict.core import remove as _remove
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class remove_test_case(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_remove_with_single_key(self):
|
||
|
|
d = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
'c': 3,
|
||
|
|
}
|
||
|
|
_remove(d, 'c')
|
||
|
|
r = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
}
|
||
|
|
self.assertEqual(d, r)
|
||
|
|
|
||
|
|
def test_remove_with_multiple_keys_as_args(self):
|
||
|
|
d = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
'c': 3,
|
||
|
|
'd': 4,
|
||
|
|
'e': 5,
|
||
|
|
}
|
||
|
|
_remove(d, 'c', 'd', 'e')
|
||
|
|
r = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
}
|
||
|
|
self.assertEqual(d, r)
|
||
|
|
|
||
|
|
def test_remove_with_multiple_keys_as_list(self):
|
||
|
|
d = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
'c': 3,
|
||
|
|
'd': 4,
|
||
|
|
'e': 5,
|
||
|
|
}
|
||
|
|
_remove(d, ['c', 'd', 'e'])
|
||
|
|
r = {
|
||
|
|
'a': 1,
|
||
|
|
'b': 2,
|
||
|
|
}
|
||
|
|
self.assertEqual(d, r)
|