Ejemplo n.º 1
0
 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'])
Ejemplo n.º 2
0
 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'])
Ejemplo n.º 3
0
 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)