718 integration test

[AntonZogk]

Feat test main

Summary

Adding a testing for main.py , this is to check if all the methods are integrated together, also fixed some minor issues in main package which caused problems with main

Type of Change

• Bug fix
• New feature
• Breaking change
• Documentation update
• Other (please describe):

Checklists

This pull request meets the following requirements:

Creator Checklist

• Installable with all dependencies recorded
• Runs without error
• Follows PEP8 and project-specific conventions
• Appropriate use of comments, for example, no descriptive comments
• Functions documented using Numpy style docstrings
• Assumptions and decisions log considered and updated if appropriate
• Unit tests have been updated to cover essential functionality for a reasonable range of inputs and conditions
• Other forms of testing such as end-to-end and user-interface testing have been considered and updated as required

If you feel some of these conditions do not apply for this pull request, please
add a comment to explain why.

Reviewer Checklist

• Test suite passes (locally as a minimum)
• Peer reviewed with review recorded

Additional Information

Please provide any additional information or context that would help the reviewer understand the changes in this pull request.

Related Issues

Link any related issues or pull requests here.