changed import statements in *tests.py files