Added test framework.

Moving the function definition on the same line would allow you to start the function body with just 2 spaces of indentation. I think that would look cleaner.