change: add queryLineageResult visualizer unit test

Author: Yi-Ting Lee

tests/integ/sagemaker/lineage/test_lineage_visualize.py

@@ -57,10 +57,6 @@ def test_wide_graph_visualize(sagemaker_session):
         lq_result = lq.query(start_arns=[wide_graph_root_arn])
         lq_result.visualize(path="wideGraph.html")
 
-        print("vertex len = ")
-        print(len(lq_result.vertices))
-        assert False
-
     except Exception as e:
         print(e)
         assert False

tests/unit/sagemaker/lineage/test_query.py

@@ -563,8 +563,8 @@ def test_query_lineage_result_str(sagemaker_session):
     )
 
     response_str = query_response.__str__()
-    pattern = "Mock id='\d*'"
-    replace = "Mock id=''"
+    pattern = r"Mock id='\d*'"
+    replace = r"Mock id=''"
     response_str = re.sub(pattern, replace, response_str)
 
     assert (