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