Add tests using mocker to main.py

[sweep-nightly]

Description

This PR adds tests to the main.py file using a mocking library. The tests ensure that the code in main.py is working as expected.

Summary of Changes

• Created a new file named tests/test_main.py to contain the tests.
• Added import statements for the necessary testing libraries, such as pytest and unittest.mock.
• Imported the necessary functions and classes from main.py into the test file.
• Implemented test functions for each function or class in main.py, using mock objects to simulate external dependencies.
• Decorated test functions with @pytest.mark.parametrize and @pytest.mark.asyncio as needed.
• Added docstrings to all functions and classes in main.py to describe their purpose, inputs, outputs, and side effects.
• Ensured proper formatting, indentation, and naming conventions in main.py.
• Organized and grouped imports at the top of main.py, and removed any unused imports or variables.
• Added comments throughout the code to provide explanations.

Please review and merge this PR to incorporate the tests and improve the code quality in main.py.

Fixes #10.

---

🎉 Latest improvements to Sweep:

• Sweep can now passively improve your repository! Check out Rules to learn more.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly]

Rollback Files For Sweep

• Rollback changes to src/main.py
• Rollback changes to tests/test_main.py

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.