Add some unit tests

[diatrevolo]

Is your feature request related to a problem? Please describe.
Need some tests, let's be clear.

Describe the solution you'd like
Write the tests!

Describe alternatives you've considered
N/A

Additional context
N/A