aboutsummaryrefslogtreecommitdiff
path: root/tests/aiet/test_main.py
blob: f2ebae20d6aa557b7a3fbd895c2a2f6734768ad1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Module for testing AIET main.py."""
from typing import Any
from unittest.mock import MagicMock

from aiet import main


def test_main(monkeypatch: Any) -> None:
    """Test main entry point function."""
    with monkeypatch.context() as mock_context:
        mock = MagicMock()
        mock_context.setattr(main, "cli", mock)
        main.main()
        mock.assert_called_once()