Add unit test feature to documentation and sample file