def test_set_optimization_precision_on_empty_config(
self,
mocked_load_precisions_config: MagicMock,
) -> None:
"""Test set_optimization_precision."""
mocked_load_precisions_config.return_value = {
"framework_foo": [
{
"name": "precision1",
},
{
"name": "precision2",
},
{
"name": "precision3",
},
],
"framework_bar": [
{
"name": "precision1",
},
],
}
config = Config()
config.set_optimization_precision("framework_foo", "precision2")
self.assertEqual("precision2", config.graph_optimization.precisions)
mocked_load_precisions_config.assert_called_once()
def test_set_optimization_precision_to_unknown_framework(
self,
mocked_load_precisions_config: MagicMock,
) -> None:
"""Test set_optimization_precision."""
mocked_load_precisions_config.return_value = {
"framework_foo": [
{
"name": "precision1",
},
{
"name": "precision2",
},
{
"name": "precision3",
},
],
"framework_bar": [
{
"name": "precision1",
},
],
}
config = Config(self.predefined_config)
with self.assertRaisesRegex(
ClientErrorException,
"Precision precision1 is not supported "
"in graph optimization for framework framework_baz.",
):
config.set_optimization_precision("framework_baz", "precision1")
mocked_load_precisions_config.assert_called_once()