Testing for ML Codebase

[manascb1344]

Introduce comprehensive testing to the machine learning codebase using Python.

• Increase the number of unit tests within the codebase to ensure that individual components function as expected.
• Add integration tests to verify that different components of the machine learning system work together seamlessly.
• Utilize Pytest as the testing framework to run the tests, providing a robust and flexible testing environment.
• Implement continuous testing practices to ensure that new changes do not introduce regressions or break existing functionality.