mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 12:55:00 +00:00
31 lines
978 B
Python
31 lines
978 B
Python
import unittest
|
|
from unittest.mock import patch
|
|
import sys
|
|
import os
|
|
|
|
# Add the parent directory to the Python path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
|
|
|
|
from aider.main import main
|
|
|
|
class TestBrowser(unittest.TestCase):
|
|
@patch('aider.main.launch_gui')
|
|
def test_browser_flag_imports_streamlit(self, mock_launch_gui):
|
|
# Run main with --browser and --yes flags
|
|
main(['--browser', '--yes'])
|
|
|
|
# Check that launch_gui was called
|
|
mock_launch_gui.assert_called_once()
|
|
|
|
# Try to import streamlit
|
|
try:
|
|
import streamlit
|
|
streamlit_imported = True
|
|
except ImportError:
|
|
streamlit_imported = False
|
|
|
|
# Assert that streamlit was successfully imported
|
|
self.assertTrue(streamlit_imported, "Streamlit should be importable after running with --browser flag")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|