From 392a0471ed19d71a226442270127a9414ccc7908 Mon Sep 17 00:00:00 2001 From: Pierre-Yves Chibon Date: Tue, 24 Jul 2018 17:39:58 +0200 Subject: [PATCH] Fix tests for the drop the unversion python Signed-off-by: Pierre-Yves Chibon --- test.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test.py b/test.py index 68ef40c..71994fa 100644 --- a/test.py +++ b/test.py @@ -3007,7 +3007,10 @@ class ExecutionContextTests(unittest.TestCase): import sh _sh = sh() omg = _sh - from omg import python + if IS_PY2: + from omg import python2 + else: + from omg import python3 def test_importer_only_works_with_sh(self): def unallowed_import(): -- 2.14.4