Remove pytest fixture and fix test_tag/s method

Author: stisac

tests/integ/sagemaker/lineage/conftest.py

@@ -33,11 +33,6 @@
 SLEEP_TIME_SECONDS = 1
 
 
-@pytest.fixture()
-def sagemaker_session(boto_session):
-    return Session(boto_session=boto_session)
-
-
 @pytest.fixture
 def action_obj(sagemaker_session):
     obj = action.Action.create(

tests/integ/sagemaker/lineage/test_action.py

@@ -90,8 +90,10 @@ def test_tag(action_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags[0] == tag
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags[0] == tag
 
 
 def test_tags(action_obj, sagemaker_session):
@@ -104,5 +106,7 @@ def test_tags(action_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags == tags
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags == tags

tests/integ/sagemaker/lineage/test_artifact.py

@@ -99,8 +99,10 @@ def test_tag(artifact_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags[0] == tag
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags[0] == tag
 
 
 def test_tags(artifact_obj, sagemaker_session):
@@ -113,5 +115,7 @@ def test_tags(artifact_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags == tags
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags == tags

tests/integ/sagemaker/lineage/test_context.py

@@ -88,8 +88,10 @@ def test_tag(context_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags[0] == tag
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags[0] == tag
 
 
 def test_tags(context_obj, sagemaker_session):
@@ -102,5 +104,7 @@ def test_tags(context_obj, sagemaker_session):
         )["Tags"]
         if actual_tags:
             break
-    assert len(actual_tags) == 1
-    assert actual_tags == tags
+
+    if len(actual_tags) == 1:
+        assert len(actual_tags) == 1
+        assert actual_tags == tags