Added a simple unit tests to showcase how the library works instead of main.go

Signed-off-by: Toon Schoenmakers <nighteyes1993@gmail.com>