def test_get_leaf_nested_dict_not_leaf(self): with self.assertRaisesRegex(KeyError, 'The value extracted with keys.*'): d = {'a': {'i': {'x': 5}}} train_utils.get_leaf_nested_dict(d, ['a', 'i'])
def test_get_leaf_nested_dict_path_not_exist_meets_leaf(self): with self.assertRaisesRegex(KeyError, 'Path not exist while traversing .*'): d = {'a': {'i': 5}} train_utils.get_leaf_nested_dict(d, ['a', 'i', 'z'])
def test_get_leaf_nested_dict(self): d = {'a': {'i': {'x': 5}}} self.assertEqual(train_utils.get_leaf_nested_dict(d, ['a', 'i', 'x']), 5)