Run on pytest 5 and 6 - 4 is no compatible

jerry-git · Jun 8, 2021 · cb87064 · cb87064
1 parent 41fe775
commit cb87064
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -33,7 +33,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: [ "3.6", "3.7", "3.8", "3.9" ]
-        pytest-version: [ "4", "5", "6" ]
+        pytest-version: [ "5", "6" ]
     steps:
       - name: Check out repository
         uses: actions/checkout@v2

diff --git a/src/pytest_split/plugin.py b/src/pytest_split/plugin.py
@@ -10,7 +10,9 @@
     from typing import List, Tuple, Optional, Union
 
     from _pytest import nodes
-    from _pytest.config import Config, ExitCode
+
+    from _pytest.main import ExitCode
+    from _pytest.config import Config
     from _pytest.config.argparsing import Parser
 
 # Ugly hack for freezegun compatibility: https://github.com/spulec/freezegun/issues/286

diff --git a/tests/test_plugin.py b/tests/test_plugin.py
@@ -3,7 +3,7 @@
 import os
 
 import pytest
-from _pytest.config import ExitCode
+from _pytest.main import ExitCode
 
 pytest_plugins = ["pytester"]