build_tools/benchmarks/common/common_arguments_test.py

#!/usr/bin/env python3
# Copyright 2022 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

import unittest
import shutil
import tempfile

from common.common_arguments import build_common_argument_parser


class CommonArgumentsTest(unittest.TestCase):

  def test_build_common_argument_parser(self):
    with tempfile.TemporaryDirectory() as tempdir:
      arg_parser = build_common_argument_parser()
      arg_parser.parse_args([
          "--normal_benchmark_tool_dir=" + tempdir,
          "--traced_benchmark_tool_dir=" + tempdir,
          "--trace_capture_tool=" + shutil.which("ls"), "."
      ])

  def test_build_common_argument_parser_check_build_dir(self):
    arg_parser = build_common_argument_parser()
    with self.assertRaises(SystemExit):
      arg_parser.parse_args(["nonexistent"])

  def test_build_common_argument_parser_check_normal_benchmark_tool(self):
    arg_parser = build_common_argument_parser()
    with self.assertRaises(SystemExit):
      arg_parser.parse_args(["--normal_benchmark_tool_dir=nonexistent", "."])

  def test_build_common_argument_parser_check_traced_benchmark_tool(self):
    arg_parser = build_common_argument_parser()
    with self.assertRaises(SystemExit):
      arg_parser.parse_args(["--traced_benchmark_tool_dir=nonexistent", "."])

  def test_build_common_argument_parser_check_trace_capture_tool(self):
    arg_parser = build_common_argument_parser()
    with self.assertRaises(SystemExit):
      arg_parser.parse_args(["--trace_capture_tool=nonexistent", "."])


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