Add unit tests

[johnslavik]

Describe the use case of a new functionality

Software without unit tests is like a child without shoes.

Prone to pain.

Example Use

(Does not apply)

Additional context

No response