Convert to unit test
Merge pull request #2 from rhattersley/unit

Convert to rhattersley unit test