Changed import statement

changed import to use test.py in tests

Change-Id: If0642797724d3edbfed772417de3708785f71a8f