Clean up test_extensions.py imports

Remove unnessesary imports
Import only modules

Change-Id: I4988714e3ae31ce77f3a2c1e27e81dd6258c7546
Merge "Clean up test_extensions.py imports"