Moves test to test directory.
Minor PEP8 fix.