|
36 | 36 | create_document_dataset, |
37 | 37 | create_image_dataset, |
38 | 38 | ) |
39 | | -from tests.integration.feature_repos.integration_test_repo_config import ( # noqa: E402 |
40 | | - IntegrationTestRepoConfig, |
41 | | -) |
42 | | -from tests.integration.feature_repos.repo_configuration import ( # noqa: E402 |
43 | | - AVAILABLE_OFFLINE_STORES, |
44 | | - AVAILABLE_ONLINE_STORES, |
45 | | - OFFLINE_STORE_TO_PROVIDER_CONFIG, |
46 | | - Environment, |
47 | | - TestData, |
48 | | - construct_test_environment, |
49 | | - construct_universal_feature_views, |
50 | | - construct_universal_test_data, |
51 | | -) |
52 | | -from tests.integration.feature_repos.universal.data_sources.file import ( # noqa: E402 |
53 | | - FileDataSourceCreator, |
54 | | -) |
55 | | -from tests.integration.feature_repos.universal.entities import ( # noqa: E402 |
56 | | - customer, |
57 | | - driver, |
58 | | - location, |
59 | | -) |
60 | | -from tests.utils.auth_permissions_util import default_store |
| 39 | +try: |
| 40 | + from tests.integration.feature_repos.integration_test_repo_config import ( # noqa: E402 |
| 41 | + IntegrationTestRepoConfig, |
| 42 | + ) |
| 43 | + from tests.integration.feature_repos.repo_configuration import ( # noqa: E402 |
| 44 | + AVAILABLE_OFFLINE_STORES, |
| 45 | + AVAILABLE_ONLINE_STORES, |
| 46 | + OFFLINE_STORE_TO_PROVIDER_CONFIG, |
| 47 | + Environment, |
| 48 | + TestData, |
| 49 | + construct_test_environment, |
| 50 | + construct_universal_feature_views, |
| 51 | + construct_universal_test_data, |
| 52 | + ) |
| 53 | + from tests.integration.feature_repos.universal.data_sources.file import ( # noqa: E402 |
| 54 | + FileDataSourceCreator, |
| 55 | + ) |
| 56 | + from tests.integration.feature_repos.universal.entities import ( # noqa: E402 |
| 57 | + customer, |
| 58 | + driver, |
| 59 | + location, |
| 60 | + ) |
| 61 | +except ModuleNotFoundError: |
| 62 | + IntegrationTestRepoConfig = None # type: ignore[assignment] |
| 63 | + Environment = None # type: ignore[assignment] |
| 64 | + TestData = None # type: ignore[assignment] |
| 65 | + AVAILABLE_OFFLINE_STORES = None # type: ignore[assignment] |
| 66 | + AVAILABLE_ONLINE_STORES = None # type: ignore[assignment] |
| 67 | + OFFLINE_STORE_TO_PROVIDER_CONFIG = None # type: ignore[assignment] |
| 68 | + construct_test_environment = None # type: ignore[assignment] |
| 69 | + construct_universal_feature_views = None # type: ignore[assignment] |
| 70 | + construct_universal_test_data = None # type: ignore[assignment] |
| 71 | + FileDataSourceCreator = None # type: ignore[assignment] |
| 72 | + customer = None # type: ignore[assignment] |
| 73 | + driver = None # type: ignore[assignment] |
| 74 | + location = None # type: ignore[assignment] |
| 75 | +try: |
| 76 | + from tests.utils.auth_permissions_util import default_store |
| 77 | +except ModuleNotFoundError: |
| 78 | + default_store = None # type: ignore[assignment] |
61 | 79 | from tests.utils.http_server import check_port_open, free_port # noqa: E402 |
62 | | -from tests.utils.ssl_certifcates_util import ( |
63 | | - combine_trust_stores, |
64 | | - create_ca_trust_store, |
65 | | - generate_self_signed_cert, |
66 | | -) |
| 80 | +try: |
| 81 | + from tests.utils.ssl_certifcates_util import ( |
| 82 | + combine_trust_stores, |
| 83 | + create_ca_trust_store, |
| 84 | + generate_self_signed_cert, |
| 85 | + ) |
| 86 | +except ModuleNotFoundError: |
| 87 | + combine_trust_stores = None # type: ignore[assignment] |
| 88 | + create_ca_trust_store = None # type: ignore[assignment] |
| 89 | + generate_self_signed_cert = None # type: ignore[assignment] |
67 | 90 |
|
68 | 91 | logger = logging.getLogger(__name__) |
69 | 92 |
|
|
85 | 108 |
|
86 | 109 |
|
87 | 110 | def pytest_configure(config): |
88 | | - if platform in ["darwin", "windows"]: |
| 111 | + if platform == "darwin" or platform.startswith("win"): |
89 | 112 | multiprocessing.set_start_method("spawn", force=True) |
90 | 113 | else: |
91 | 114 | multiprocessing.set_start_method("fork") |
|
0 commit comments