Fix relative imports in tests package.
Merge pull request #4 from ajdawson/fix-relative-imports

Fix relative imports in tests package.