Fix linter errors in test files

What does this MR do and why?

This MR is to add fixes for linter errors that appear in test files. It's based on top of !101 (merged)

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.