Change to public exposed pytest types

Author: ubaumann

nuts/context.py

@@ -1,7 +1,7 @@
 """Provide necessary information that is needed for a specific test."""
 import pathlib
 from typing import Any, Callable, Optional, Dict
-from _pytest.config import Config
+from pytest import Config
 
 from nornir import InitNornir
 from nornir.core import Nornir

nuts/plugin.py

@@ -3,12 +3,12 @@
 from pathlib import Path
 
 import pytest
-from _pytest.config.argparsing import Parser
-from _pytest.main import Session
-from _pytest.nodes import Collector
-from _pytest.python import Metafunc
-from _pytest.fixtures import FixtureRequest
-from _pytest.config import Config
+from pytest import Parser
+from pytest import Session
+from pytest import Collector
+from pytest import Metafunc
+from pytest import FixtureRequest
+from pytest import Config
 
 from nuts.context import NutsContext
 from nuts.context import NornirNutsContext

nuts/yamlloader.py

@@ -6,12 +6,12 @@
 from importlib import util
 from typing import Iterable, Union, Any, Optional, List, Set, Dict, Tuple
 
-import pytest
 import yaml
+import pytest
+from pytest import Item, Collector
+from pytest import Metafunc
 from _pytest.mark import ParameterSet
 from _pytest.nodes import Node
-from _pytest.nodes import Item, Collector
-from _pytest.python import Metafunc
 
 # from pytest import Metafunc, Item, Collector
 

tests/base_tests/conftest.py

@@ -2,7 +2,7 @@
 
 import yaml
 import pytest
-from _pytest.fixtures import FixtureRequest
+from pytest import FixtureRequest
 from napalm.base.exceptions import ConnectionException
 from nornir.core.task import AggregatedResult