Add basic unit test

Would be nice if the example could also cover a simple and basic unit test...