example of how to write unit tests