Streamline the import of modules in tests

[surbhigoel77]

In current codebase, the import of modules in test files is inconsistent and needs to be modified in a way that the required modules are imported within the testcases instead of importing all modules at once, so that failure of one import does not impact other testcases.