Update integ test for checking Python version (#189)

* Update integ test for checking Python version

* remove emacs backup file

* remove extraneous code

* Python 2 compatibility

Author: laurenyu

test/integration/docker_utils.py

@@ -1,4 +1,4 @@
-# Copyright 2017-2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# Copyright 2017-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"). You
 # may not use this file except in compliance with the License. A copy of
@@ -18,8 +18,6 @@
 import pytest
 from sagemaker.tensorflow import TensorFlow
 
-from test.integration.docker_utils import Container
-
 RESOURCE_PATH = os.path.join(os.path.dirname(__file__), '..', '..', 'resources')
 TF_CHECKPOINT_FILES = ['graph.pbtxt', 'model.ckpt-0.index', 'model.ckpt-0.meta']
 
@@ -32,10 +30,23 @@ def py_full_version(py_version):
         return '3.6'
 
 
-def test_py_versions(docker_image, processor, py_full_version):
-    with Container(docker_image, processor) as c:
-        output = c.execute_command(['python', '--version'])
-        assert output.strip().startswith('Python {}'.format(py_full_version))
+def test_py_versions(sagemaker_local_session, docker_image, py_full_version, framework_version, tmpdir):
+    output_path = 'file://{}'.format(tmpdir)
+    run_tf_training(script=os.path.join(RESOURCE_PATH, 'test_py_version', 'entry.py'),
+                    instance_type='local',
+                    instance_count=1,
+                    sagemaker_local_session=sagemaker_local_session,
+                    docker_image=docker_image,
+                    framework_version=framework_version,
+                    output_path=output_path,
+                    training_data_path=None)
+
+    with tarfile.open(os.path.join(str(tmpdir), 'output.tar.gz')) as tar:
+        output_file = tar.getmember('py_version')
+        tar.extractall(path=str(tmpdir), members=[output_file])
+
+    with open(os.path.join(str(tmpdir), 'py_version')) as f:
+        assert f.read().strip() == py_full_version
 
 
 @pytest.mark.skip_gpu

test/integration/local/test_training.py

@@ -0,0 +1,22 @@
+# Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from __future__ import absolute_import
+
+import os
+import sys
+
+
+py_version = '%s.%s' % (sys.version_info.major, sys.version_info.minor)
+
+with open(os.path.join(os.environ['SM_OUTPUT_DIR'], 'py_version'), 'a') as f:
+    f.write(py_version)