def test_get_models_fails_without_domain(self) -> None:
"""Test that get_models fails when no config given."""
feeder = Feeder(data={
"config": {
"framework": "framework_foo",
},
}, )
with self.assertRaisesRegex(ClientErrorException, "Domain not set."):
feeder.get_models()
def test_get_models(
self,
mocked_load_model_config: MagicMock,
) -> None:
"""Test get_models method."""
mocked_load_model_config.return_value = {
"__help__framework_foo": "help for framework_foo",
"framework_foo": {
"__help__domain1": "help text for framework_foo/domain1",
"domain1": {},
"__help__domain2": "help text for framework_foo/domain2",
"domain2": {
"__help__model1": "help for model 1",
"model1": {},
"__help__model2": "help for model 2",
"model2": {},
"__help__model3": "help for model 3",
"model3": {},
},
"__help__domain3": "help text for framework_foo/domain3",
"domain3": {},
},
"__help__framework_bar": "help for framework_bar",
"framework_bar": {
"__help__domain1": "help text for framework_bar/domain1",
"domain1": {},
},
"__help__framework_baz": "help for framework_baz",
"framework_baz": {
"__help__domain1": "help text for framework_baz/domain1",
"domain1": {},
},
}
expected = [
{
"name": "model1",
"help": "help for model 1",
},
{
"name": "model2",
"help": "help for model 2",
},
{
"name": "model3",
"help": "help for model 3",
},
]
feeder = Feeder(data={
"config": {
"framework": "framework_foo",
"domain": "domain2",
},
}, )
actual = feeder.get_models()
mocked_load_model_config.assert_called_once()
self.assertEqual(expected, actual)
def test_get_models_fails_without_framework(self) -> None:
"""Test that get_models fails when no config given."""
feeder = Feeder(data={})
with self.assertRaisesRegex(ClientErrorException, "Framework not set."):
feeder.get_models()
