[FEATURE] Implement Unit-Tests

### Is your feature request related to a problem? If so, please describe.

It's hard to implement new things and fix some bugs, Unit Tests on these problems might be easy to work on

### What feature or improvement would you like to see?

Use some framework to test

### Additional context

_No response_