infra: remove assumption of Python 2 unit test runs

Author: laurenyu

tests/conftest.py

@@ -133,6 +133,11 @@ def chainer_version(request):
     return request.param
 
 
+@pytest.fixture(scope="module", params=["py2", "py3"])
+def chainer_py_version(request):
+    return request.param
+
+
 # TODO: current version fixtures are legacy fixtures that aren't useful
 # and no longer verify whether images are valid
 @pytest.fixture(

tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_framework_version.py

@@ -12,8 +12,6 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-import sys
-
 import pasta
 import pytest
 
@@ -106,13 +104,6 @@ def constructors(versions=False, image=False):
     return [ctr for template in TEMPLATES for ctr in template.constructors(versions, image)]
 
 
-@pytest.fixture(autouse=True)
-def skip_if_py2():
-    # Remove once https://github.com/aws/sagemaker-python-sdk/issues/1461 is addressed.
-    if sys.version_info.major < 3:
-        pytest.skip("v2 migration script doesn't support Python 2.")
-
-
 @pytest.fixture
 def constructors_empty():
     return constructors()

tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_tf_legacy_mode.py

@@ -12,10 +12,7 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-import sys
-
 import pasta
-import pytest
 from mock import MagicMock, patch
 
 from sagemaker.cli.compatibility.v2.modifiers import tf_legacy_mode
@@ -25,13 +22,6 @@
 REGION_NAME = "us-west-2"
 
 
-@pytest.fixture(autouse=True)
-def skip_if_py2():
-    # Remove once https://github.com/aws/sagemaker-python-sdk/issues/1461 is addressed.
-    if sys.version_info.major < 3:
-        pytest.skip("v2 migration script doesn't support Python 2.")
-
-
 def test_node_should_be_modified_tf_constructor_legacy_mode():
     tf_legacy_mode_constructors = (
         "TensorFlow(script_mode=False)",