Added pytest fixture (#309)

Co-authored-by: SK <saravsak@amazon.com>
Co-authored-by: Karim Nakad <karimnakad@gmail.com>

Author: 3 people

test/integration/conftest.py

@@ -112,6 +112,13 @@ def docker_image(docker_base_name, tag):
     return '{}:{}'.format(docker_base_name, tag)
 
 
+@pytest.fixture(autouse=True)
+def skip_py2_containers(request, tag):
+    if request.node.get_closest_marker('skip_py2_containers'):
+        if 'py2' in tag:
+            pytest.skip('Skipping python2 container with tag {}'.format(tag))
+
+
 @pytest.fixture
 def ecr_image(account_id, docker_base_name, tag, region):
     return '{}.dkr.ecr.{}.amazonaws.com/{}:{}'.format(