runtime/bindings/python/tests/flags_test.py

# Copyright 2019 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

from iree import runtime as rt
import os
import numpy as np
import unittest


class FlagsTest(unittest.TestCase):
    def setUp(self):
        os.environ["IREE_PY_RUNTIME_FLAGS"] = ""

    def testParse(self):
        # --help is always available if flags are.
        rt.flags.parse_flags("--help")

    def testParseError(self):
        with self.assertRaisesRegex(ValueError, "flag 'barbar' not recognized"):
            rt.flags.parse_flags("--barbar")

    def testEnvParse(self):
        os.environ["IREE_PY_RUNTIME_FLAGS"] = "--help"
        rt.flags._load_default_flags_from_env()

    def testEnvParseError(self):
        os.environ["IREE_PY_RUNTIME_FLAGS"] = "--barbar"
        with self.assertRaisesRegex(RuntimeError, "IREE_PY_RUNTIME_FLAGS.*barbar"):
            rt.flags._load_default_flags_from_env()


if __name__ == "__main__":
    unittest.main()