Remove pytest fixture and fix test_tag/s method

Author: stisac

tests/integ/sagemaker/lineage/test_action.py

@@ -90,10 +90,10 @@ def test_tag(action_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags[0] == tag
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert actual_tags[0] == tag
 
 
 def test_tags(action_obj, sagemaker_session):
@@ -106,7 +106,7 @@ def test_tags(action_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags == tags
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert [actual_tags[-1]] == tags

tests/integ/sagemaker/lineage/test_artifact.py

@@ -99,10 +99,10 @@ def test_tag(artifact_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags[0] == tag
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert actual_tags[0] == tag
 
 
 def test_tags(artifact_obj, sagemaker_session):
@@ -115,7 +115,7 @@ def test_tags(artifact_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags == tags
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert [actual_tags[-1]] == tags

tests/integ/sagemaker/lineage/test_context.py

@@ -88,10 +88,10 @@ def test_tag(context_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags[0] == tag
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert actual_tags[0] == tag
 
 
 def test_tags(context_obj, sagemaker_session):
@@ -104,7 +104,7 @@ def test_tags(context_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-
-    if len(actual_tags) == 1:
-        assert len(actual_tags) == 1
-        assert actual_tags == tags
+    # When sagemaker-client-config endpoint-url is passed as argument to hit beta,
+    # length of actual tags will be 2
+    assert len(actual_tags) > 0
+    assert [actual_tags[-1]] == tags