Update test PR to work with latest code, fix flake8

tests/utils.py

@@ -1,5 +1,6 @@
 import os
 
+
 def get_artifact(fname):
     """
     Return the artifact path in the artifact folder
@@ -10,9 +11,11 @@ def get_artifact(fname):
         return fpath
     return
 
+
 def get_artifact_folder():
     return os.path.join(os.path.dirname(__file__), "artifacts")
 
+
 def init_setup():
     """
     init data to have a clean state before testing