Create tests for python code

# Feature Request

## Description

Need to create tests for python code.

## Alternatives Considered

- [ ] Create Python tests
- [ ] Implement a GitHub Actions using this test

## Additional Context

For this issue, the tests should be created using pytest, but I have never created tests using it before.
