Add imports for tests.
Merge pull request #3 from metajack/test-imports

Add imports for tests.