Update unit testing section to reflect what's really going on

Also fixed the code sample