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()