def test_insert_missing_key_with_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( lambda key: 'the key is ' + key, c=33, ) d['42'] = 'e' expected = { '42': 'e', 'c': 33, } self.assertEqual(d, expected)
def test_insert_missing_key_and_no_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( a=42, b='33', ) d['c'] = 24 expected = { 'a': 42, 'b': '33', 'c': 24, } self.assertEqual(d, expected)
def test_missing_key_with_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( lambda key: 'the key is ' + key + ' !', c=33, ) actual = d['42'] self.assertEqual(actual, 'the key is 42 !') expected_dict = { '42': 'the key is 42 !', 'c': 33, } self.assertEqual(d, expected_dict)
def test_insert_existing_key_with_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( lambda key: 'the key is ' + key, c=33, d=42, ) d['c'] = 'a' expected = { 'c': 'a', 'd': 42, } self.assertEqual(d, expected)
def test_insert_existing_key_and_no_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( a=42, b='33', c='24', ) d['a'] = 'b' expected = { 'a': 'b', 'b': '33', 'c': '24', } self.assertEqual(d, expected)
def test_existing_key_with_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( lambda key: 'the key is ' + key, c=33, d=42, ) actual = d['c'] self.assertEqual(actual, 33) expected_dict = { 'c': 33, 'd': 42, } self.assertEqual(d, expected_dict)
def test_missing_key_and_no_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( a=42, b='33', ) with self.assertRaises(KeyError) as context: _ = d['c'] self.assertEqual(context.exception.args[0], 'c') expected_dict = { 'a': 42, 'b': '33' } self.assertEqual(d, expected_dict)
def test_existing_key_and_no_default_factory(self): d = key_aware_defaultdict.KeyAwareDefaultDict( a=42, b='33', c='24', ) actual = d['a'] self.assertEqual(actual, 42) expected_dict = { 'a': 42, 'b': '33', 'c': '24' } self.assertEqual(d, expected_dict)