Python Feeder.get_precisionsの例

コード例 #1

ファイル: test_params_feeder.py プロジェクト: vuiseng9/lpot

    def test_get_precisions_for_missing_framework(
        self,
        mocked_load_precisions_config: MagicMock,
    ) -> None:
        """Test get_precisions function."""
        mocked_load_precisions_config.return_value = {
            "framework_foo": [
                {
                    "foo": {}
                },
                {
                    "bar": {}
                },
                {
                    "baz": {}
                },
            ],
            "framework_bar": [
                {
                    "foo": {}
                },
            ],
        }

        feeder = Feeder(data={})

        with self.assertRaisesRegex(ClientErrorException,
                                    "Framework not set."):
            feeder.get_precisions()

コード例 #2

ファイル: test_params_feeder.py プロジェクト: intel/lpot

    def test_get_precisions_for_unknown_framework(
        self,
        mocked_load_precisions_config: MagicMock,
    ) -> None:
        """Test get_precisions function."""
        mocked_load_precisions_config.return_value = {
            "framework_foo": [
                {"foo": {}},
                {"bar": {}},
                {"baz": {}},
            ],
            "framework_bar": [
                {"foo": {}},
            ],
        }

        feeder = Feeder(
            data={
                "config": {
                    "framework": "framework_baz",
                },
            },
        )
        actual = feeder.get_precisions()

        self.assertEqual([], actual)