fixed imports in tests for python 3 compatibility