Add unit test implementation

[stefan505]

Is your feature request related to a problem? Please describe.
Currently this library has no unit tests implemented.

Describe the solution you'd like
Advocate for unit tests to be implemented & run on PRs.

Describe alternatives you've considered
N/A

Additional context
Align implementation with other Omni repos.