def test_resolve_bare_default_args(self): test_obj = yaml.load(""" a: !DummyClass arg1: !DummyClass2 {} b: !DummyClass3 {} """) YamlPreloader._resolve_bare_default_args(test_obj) self.assertIsInstance(test_obj["a"].arg1.arg1, DummyClass) self.assertIsInstance(test_obj["b"].arg1, DummyClass2) self.assertIsInstance(test_obj["b"].arg1.arg1, DummyClass)
def test_resolve_bare_default_args_illegal(self): test_obj = yaml.load(""" a: !DummyClassForgotBare {} """) with self.assertRaises(ValueError): YamlPreloader._resolve_bare_default_args(test_obj)