Changed test imports.
Added a test to check against known correct output.