language_version works for python.

asottile · asottile · commit 4b98b39cea3c · 2014-06-02T07:26:44.000-07:00
diff --git a/pre_commit/languages/python.py b/pre_commit/languages/python.py
@@ -23,7 +23,10 @@ def install_environment(repo_cmd_runner, version='default'):
 
     # Install a virtualenv
     with clean_path_on_failure(repo_cmd_runner.path(ENVIRONMENT_DIR)):
-        repo_cmd_runner.run(['virtualenv', '{{prefix}}{0}'.format(ENVIRONMENT_DIR)])
+        venv_cmd = ['virtualenv', '{{prefix}}{0}'.format(ENVIRONMENT_DIR)]
+        if version != 'default':
+            venv_cmd.extend(['-p', version])
+        repo_cmd_runner.run(venv_cmd)
         with in_env(repo_cmd_runner) as env:
             env.run('cd {prefix} && pip install .')
 
diff --git a/setup.py b/setup.py
@@ -36,6 +36,7 @@
         'plumbum',
         'pyyaml',
         'simplejson',
+        'virtualenv',
     ],
     entry_points={
         'console_scripts': [
diff --git a/tests/repository_test.py b/tests/repository_test.py
@@ -24,7 +24,6 @@ def test_run_a_python_hook(config_for_python_hooks_repo, store):
     assert ret[1] == "['/dev/null']\nHello World\n"
 
 
-@pytest.mark.xfail
 @pytest.mark.integration
 def test_run_versioned_hook(config_for_python3_hooks_repo, store):
     repo = Repository.create(config_for_python3_hooks_repo, store)
