Add pytest-based tests for Python SDK

[CoderOMaster]

Request

Create pytest-based tests for the Python SDK to ensure functionality and improve code reliability.

Details

• Develop tests for core modules and functionality in the Python SDK.
• Use pytest as the testing framework for consistency.
• Ensure coverage for existing and major new features.
• Integrate tests to run in the CI pipeline if not already applied.

Acceptance Criteria

• All critical components and features of the Python SDK are covered by pytest-based tests.
• Tests pass successfully in the CI pipeline.

---

Feel free to discuss or break this into sub-issues as needed.

[PredictiveManish]

Working on this issue.