python version fix for tests

Author: kossak

tests/functional/test_run.py

@@ -1,3 +1,5 @@
+import sys
+
 import mock
 from click.testing import CliRunner
 
@@ -31,7 +33,7 @@ def test_run_simple_file_with_args(self, get_files_patched, workspace_zip_patche
         post_patched.assert_called_with(self.url,
                                         params={'name': u'test', 'projectId': u'projectId',
                                                 'workspaceFileName': 'bar',
-                                                'command': 'python2 myscript.py a b',
+                                                'command': 'python{} myscript.py a b'.format(str(sys.version_info[0])),
                                                 'projectHandle': u'projectId',
                                                 'container': u'paperspace/tensorflow-python'},
                                         data=None,
@@ -52,7 +54,7 @@ def test_run_python_command_with_args_and_no_workspace(self, post_patched):
                                         params={'name': u'test', 'projectId': u'projectId',
                                                 'workspaceFileName': 'none',
                                                 'workspace': 'none',
-                                                'command': 'python2 -c print(foo)',
+                                                'command': 'python{} -c print(foo)'.format(str(sys.version_info[0])),
                                                 'projectHandle': u'projectId',
                                                 'container': u'paperspace/tensorflow-python'},
                                         data=None,