Add tests using mocker to main.py

[sweep-nightly]

Description

This PR adds tests using the mocker library to the main.py file. The tests cover the functionality of the transform object, the DataLoader, and the model (once it's defined). The tests are written using the pytest framework and ensure that the code in main.py is working correctly.

Summary of Changes

• Created a new test file named test_main.py in the src directory.
• Imported pytest and mocker from pytest_mock.
• Imported the necessary modules and functions/classes from main.py.
• Added tests for the transform object, the DataLoader, and the model (once it's defined).
• Used mocker to create mock objects and assert the expected outputs.

Please review and merge this PR to ensure the code in main.py is thoroughly tested and reliable.

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/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.