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