ansys-fluent-core 0.29.dev0__py3-none-any.whl → 0.29.dev2__py3-none-any.whl

ansys/fluent/tests/conftest.py

@@ -1,415 +0,0 @@
-from contextlib import nullcontext
-import functools
-import inspect
-import operator
-import os
-from pathlib import Path
-import shutil
-import sys
-
-from packaging.specifiers import SpecifierSet
-from packaging.version import Version
-import pytest
-
-import ansys.fluent.core as pyfluent
-from ansys.fluent.core.examples.downloads import download_file
-from ansys.fluent.core.utils.file_transfer_service import RemoteFileTransferStrategy
-from ansys.fluent.core.utils.fluent_version import FluentVersion
-
-sys.path.append(Path(__file__).parent / "util")
-
-
-def pytest_addoption(parser):
-    parser.addoption(
-        "--fluent-version",
-        action="store",
-        metavar="VERSION",
-        help="only run tests supported by Fluent version VERSION.",
-    )
-    parser.addoption(
-        "--nightly", action="store_true", default=False, help="run nightly tests"
-    )
-    parser.addoption(
-        "--solvermode", action="store_true", default=False, help="run solvermode tests"
-    )
-    parser.addoption(
-        "--write-fluent-journals",
-        action="store_true",
-        default=False,
-        help="Write Fluent journals for unittests",
-    )
-
-
-def pytest_runtest_setup(item):
-    if (
-        any(mark.name == "standalone" for mark in item.iter_markers())
-        and os.getenv("PYFLUENT_LAUNCH_CONTAINER") == "1"
-    ):
-        pytest.skip()
-
-    is_nightly = item.config.getoption("--nightly")
-    if not is_nightly and any(mark.name == "nightly" for mark in item.iter_markers()):
-        pytest.skip()
-
-    is_solvermode_option = item.config.getoption("--solvermode")
-    is_solvermode_path = "test_solvermode" in item.path.parts
-    if is_solvermode_option ^ is_solvermode_path:
-        pytest.skip()
-
-    version_specs = []
-    fluent_release_version = FluentVersion.current_release().value
-    for mark in item.iter_markers(name="fluent_version"):
-        spec = mark.args[0]
-        # if a test is marked as fluent_version("latest")
-        # run with dev and release Fluent versions in nightly
-        # run with release Fluent versions in PRs
-        if spec == "latest":
-            spec = (
-                f">={fluent_release_version}"
-                if is_nightly or is_solvermode_option
-                else f"=={fluent_release_version}"
-            )
-        version_specs.append(SpecifierSet(spec))
-    if version_specs:
-        combined_spec = functools.reduce(operator.and_, version_specs)
-        version = item.config.getoption("--fluent-version")
-        if version and Version(version) not in combined_spec:
-            pytest.skip()
-
-
-def pytest_collection_finish(session):
-    if session.config.getoption("--write-fluent-journals"):
-        import_path = Path(__file__).parent
-        sys.path.append(str(import_path))
-        import fluent_fixtures
-
-        launcher_args_by_fixture = {}
-        for k, v in fluent_fixtures.__dict__.items():
-            if hasattr(v, "fluent_launcher_args"):
-                launcher_args_by_fixture[k] = v.fluent_launcher_args
-        fluent_test_root = import_path / "fluent"
-        shutil.rmtree(fluent_test_root, ignore_errors=True)
-        for item in session.items:
-            skip = False
-            for mark in item.iter_markers(name="skip"):
-                skip = True
-            for mark in item.iter_markers(name="fluent_version"):
-                spec = mark.args[0]
-                # TODO: Support older versions
-                if not (
-                    spec == "latest"
-                    or Version(FluentVersion.current_dev().value) in SpecifierSet(spec)
-                ):
-                    skip = True
-            if skip:
-                continue
-            fluent_test_dir = fluent_test_root / item.module.__name__ / item.name
-            fluent_test_config = fluent_test_dir / "test.yaml"
-            fluent_test_file = fluent_test_dir / "test.py"
-            launcher_args = ""
-            parameters = inspect.signature(item.function).parameters
-            parameter_set = {p for p in parameters}
-            if not (parameter_set & set(launcher_args_by_fixture.keys())):
-                # Skipping as unittest doesn't use fluent fixture
-                continue
-            for param in parameters:
-                if param not in dir(fluent_fixtures):
-                    print(f"Skipping {item.nodeid} because of missing fixture {param}")
-                    skip = True
-                    break
-            if skip:
-                continue
-            for param in parameters:
-                if param in launcher_args_by_fixture:
-                    launcher_args = launcher_args_by_fixture[param]
-                    break
-            fluent_test_dir.mkdir(parents=True, exist_ok=True)
-            with open(fluent_test_config, "w") as f:
-                f.write(f"launcher_args: {launcher_args}\n")
-            with open(fluent_test_file, "w") as f:
-                f.write("import sys\n")
-                f.write('sys.path.append("/testing")\n')
-                f.write(
-                    f"from {item.module.__name__} import {item.name}  # noqa: E402\n"
-                )
-                f.write("from fluent_fixtures import (  # noqa: E402\n")
-                for param in parameters:
-                    f.write(f"    {param},\n")
-                f.write(")\n")
-                f.write("\n")
-                f.write(f"{item.name}(")
-                f.write(", ".join([f"{p}(globals())" for p in parameters]))
-                f.write(")\n")
-                f.write("exit()\n")
-            print(f"Written {fluent_test_file}")
-        session.items = []
-        session.testscollected = 0
-
-
-@pytest.fixture(autouse=True)
-def run_before_each_test(
-    monkeypatch: pytest.MonkeyPatch, request: pytest.FixtureRequest
-) -> None:
-    monkeypatch.setenv("PYFLUENT_TEST_NAME", request.node.name)
-    monkeypatch.setenv("PYFLUENT_CODEGEN_SKIP_BUILTIN_SETTINGS", "1")
-    pyfluent.CONTAINER_MOUNT_SOURCE = pyfluent.EXAMPLES_PATH
-    pyfluent.CONTAINER_MOUNT_TARGET = pyfluent.EXAMPLES_PATH
-
-
-class Helpers:
-    """Can be reused to provide helper methods to tests."""
-
-    def __init__(self, monkeypatch: pytest.MonkeyPatch):
-        self.monkeypatch = monkeypatch
-
-    def mock_awp_vars(self, version=None):
-        """Activates env vars for Fluent versions up to specified version, deactivates
-        env vars for versions newer than specified."""
-        if not version:
-            version = FluentVersion.current_release()
-        elif not isinstance(version, FluentVersion):
-            version = FluentVersion(version)
-        self.monkeypatch.delenv("PYFLUENT_FLUENT_ROOT", raising=False)
-        for fv in FluentVersion:
-            if fv <= version:
-                self.monkeypatch.setenv(fv.awp_var, f"ansys_inc/{fv.name}")
-            else:
-                self.monkeypatch.delenv(fv.awp_var, raising=False)
-
-    def delete_all_awp_vars(self):
-        for fv in FluentVersion:
-            self.monkeypatch.delenv(fv.awp_var, raising=False)
-
-
-@pytest.fixture
-def helpers(monkeypatch):
-    return Helpers(monkeypatch)
-
-
-pytest.wont_raise = nullcontext
-
-
-def pytest_sessionfinish(session, exitstatus):
-    if exitstatus == 5:
-        session.exitstatus = 0
-
-
-# tests_by_fixture = defaultdict(list)
-
-
-# def pytest_collection_finish(session):
-#     for k, v in sorted(tests_by_fixture.items(), key=lambda t: len(t[1]), reverse=True):
-#         print(k, len(v))
-
-
-# def pytest_itemcollected(item):
-#     if not item.nodeid.startswith("tests/test_solvermode/"):
-#         for fixture in item.fixturenames:
-#             tests_by_fixture[fixture].append(item.nodeid)
-
-
-@pytest.fixture
-def mixing_elbow_geometry_filename():
-    return download_file(
-        file_name="mixing_elbow.pmdb", directory="pyfluent/mixing_elbow"
-    )
-
-
-@pytest.fixture
-def exhaust_system_geometry_filename():
-    return download_file(
-        file_name="exhaust_system.fmd", directory="pyfluent/exhaust_system"
-    )
-
-
-def create_session(**kwargs):
-    if pyfluent.USE_FILE_TRANSFER_SERVICE:
-        file_transfer_service = RemoteFileTransferStrategy()
-        container_dict = {"mount_source": file_transfer_service.MOUNT_SOURCE}
-        return pyfluent.launch_fluent(
-            container_dict=container_dict,
-            file_transfer_service=file_transfer_service,
-            **kwargs,
-        )
-    else:
-        return pyfluent.launch_fluent(**kwargs)
-
-
-@pytest.fixture
-def new_meshing_session():
-    meshing = create_session(mode=pyfluent.FluentMode.MESHING)
-    yield meshing
-    meshing.exit()
-
-
-@pytest.fixture
-def new_pure_meshing_session():
-    meshing = create_session(mode=pyfluent.FluentMode.PURE_MESHING)
-    yield meshing
-    meshing.exit()
-
-
-@pytest.fixture
-def watertight_workflow_session(new_meshing_session):
-    new_meshing_session.workflow.InitializeWorkflow(WorkflowType="Watertight Geometry")
-    return new_meshing_session
-
-
-@pytest.fixture
-def fault_tolerant_workflow_session(new_meshing_session):
-    new_meshing_session.workflow.InitializeWorkflow(
-        WorkflowType="Fault-tolerant Meshing"
-    )
-    return new_meshing_session
-
-
-@pytest.fixture
-def mixing_elbow_watertight_pure_meshing_session(
-    new_pure_meshing_session, mixing_elbow_geometry_filename
-):
-    meshing = new_pure_meshing_session
-    meshing.workflow.InitializeWorkflow(WorkflowType="Watertight Geometry")
-    meshing.workflow.TaskObject["Import Geometry"].Arguments = dict(
-        FileName=mixing_elbow_geometry_filename, LengthUnit="in"
-    )
-
-    return meshing
-
-
-@pytest.fixture
-def new_solver_session():
-    solver = create_session()
-    yield solver
-    solver.exit()
-
-
-@pytest.fixture
-def new_solver_session_t4():
-    solver = create_session(processor_count=4)
-    yield solver
-    solver.exit()
-
-
-@pytest.fixture
-def new_solver_session_sp():
-    solver = create_session(precision="single")
-    yield solver
-    solver.exit()
-
-
-@pytest.fixture
-def new_solver_session_2d():
-    solver = create_session(dimension=2)
-    yield solver
-    solver.exit()
-
-
-@pytest.fixture
-def static_mixer_settings_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file("Static_Mixer_main.cas.h5", "pyfluent/static_mixer")
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@pytest.fixture
-def static_mixer_case_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file("Static_Mixer_main.cas.h5", "pyfluent/static_mixer")
-    solver.file.read(file_type="case", file_name=case_name)
-    return solver
-
-
-@pytest.fixture
-def mixing_elbow_settings_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file("mixing_elbow.cas.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@pytest.fixture
-def mixing_elbow_case_session_t4(new_solver_session_t4):
-    solver = new_solver_session_t4
-    case_name = download_file("mixing_elbow.cas.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(file_type="case", file_name=case_name)
-    return solver
-
-
-@pytest.fixture
-def mixing_elbow_case_data_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file("mixing_elbow.cas.h5", "pyfluent/mixing_elbow")
-    download_file("mixing_elbow.dat.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(file_type="case-data", file_name=case_name)
-    return solver
-
-
-@pytest.fixture
-def mixing_elbow_param_case_data_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file("elbow_param.cas.h5", "pyfluent/mixing_elbow")
-    download_file("elbow_param.dat.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(file_type="case-data", file_name=case_name)
-    return solver
-
-
-@pytest.fixture
-def disk_settings_session(new_solver_session_2d):
-    solver = new_solver_session_2d
-    case_name = download_file("disk.cas.h5", "pyfluent/rotating_disk")
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@pytest.fixture
-def disk_case_session(new_solver_session_2d):
-    solver = new_solver_session_2d
-    case_name = download_file("disk.cas.h5", "pyfluent/rotating_disk")
-    solver.file.read(file_type="case", file_name=case_name)
-    return solver
-
-
-@pytest.fixture
-def periodic_rot_settings_session(new_solver_session):
-    solver = new_solver_session
-    case_name = download_file(
-        "periodic_rot.cas.h5",
-        "pyfluent/periodic_rot",
-    )
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@pytest.fixture
-def disable_datamodel_cache(monkeypatch: pytest.MonkeyPatch):
-    monkeypatch.setattr(pyfluent, "DATAMODEL_USE_STATE_CACHE", False)
-
-
-@pytest.fixture(params=["old", "new"])
-def datamodel_api_version_all(request, monkeypatch: pytest.MonkeyPatch) -> None:
-    if request.param == "new":
-        monkeypatch.setenv("REMOTING_NEW_DM_API", "1")
-        monkeypatch.setenv("REMOTING_MAPPED_NEW_DM_API", "1")
-
-
-@pytest.fixture
-def datamodel_api_version_new(monkeypatch: pytest.MonkeyPatch) -> None:
-    monkeypatch.setenv("REMOTING_NEW_DM_API", "1")
-    monkeypatch.setenv("REMOTING_MAPPED_NEW_DM_API", "1")

ansys/fluent/tests/fluent_fixtures.py

@@ -1,195 +0,0 @@
-from typing import Callable
-
-import pytest
-
-import ansys.fluent.core as pyfluent
-from ansys.fluent.core.data_model_cache import DataModelCache
-from ansys.fluent.core.examples import download_file
-
-
-def fluent_launcher_args(args: str):
-    def fluent_launcher_args_inner(f: Callable):
-        def wrapper(*args, **kwargs):
-            return f(*args, **kwargs)
-
-        wrapper.fluent_launcher_args = args
-        return wrapper
-
-    return fluent_launcher_args_inner
-
-
-def mixing_elbow_geometry_filename(globals):
-    return download_file(
-        file_name="mixing_elbow.pmdb", directory="pyfluent/mixing_elbow"
-    )
-
-
-def exhaust_system_geometry_filename(globals):
-    return download_file(
-        file_name="exhaust_system.fmd", directory="pyfluent/exhaust_system"
-    )
-
-
-@fluent_launcher_args("3ddp -meshing")
-def new_meshing_session(globals):
-    meshing = globals["meshing"]
-    return meshing
-
-
-@fluent_launcher_args("3ddp -meshing")
-def new_pure_meshing_session(globals):
-    return new_meshing_session(globals)
-
-
-@fluent_launcher_args("3ddp -meshing")
-def watertight_workflow_session(globals):
-    meshing = new_meshing_session(globals)
-    meshing.workflow.InitializeWorkflow(WorkflowType="Watertight Geometry")
-    return meshing
-
-
-@fluent_launcher_args("3ddp -meshing")
-def fault_tolerant_workflow_session(globals):
-    meshing = new_meshing_session(globals)
-    meshing.workflow.InitializeWorkflow(WorkflowType="Fault-tolerant Meshing")
-    return meshing
-
-
-@fluent_launcher_args("3ddp -meshing")
-def mixing_elbow_watertight_pure_meshing_session(globals):
-    meshing = new_pure_meshing_session(globals)
-    geometry_filename = mixing_elbow_geometry_filename(globals)
-    meshing.workflow.InitializeWorkflow(WorkflowType="Watertight Geometry")
-    meshing.workflow.TaskObject["Import Geometry"].Arguments = dict(
-        FileName=geometry_filename, LengthUnit="in"
-    )
-    return meshing
-
-
-@fluent_launcher_args("3ddp")
-def new_solver_session(globals):
-    solver = globals["solver"]
-    return solver
-
-
-@fluent_launcher_args("3d")
-def new_solver_session_sp(globals):
-    return new_solver_session(globals)
-
-
-@fluent_launcher_args("2ddp")
-def new_solver_session_2d(globals):
-    return new_solver_session(globals)
-
-
-@fluent_launcher_args("3ddp")
-def static_mixer_settings_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file("Static_Mixer_main.cas.h5", "pyfluent/static_mixer")
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@fluent_launcher_args("3ddp")
-def static_mixer_case_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file("Static_Mixer_main.cas.h5", "pyfluent/static_mixer")
-    solver.file.read(file_type="case", file_name=case_name)
-    return solver
-
-
-@fluent_launcher_args("3ddp")
-def mixing_elbow_settings_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file("mixing_elbow.cas.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@fluent_launcher_args("3ddp")
-def mixing_elbow_case_data_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file("mixing_elbow.cas.h5", "pyfluent/mixing_elbow")
-    download_file("mixing_elbow.dat.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(file_type="case-data", file_name=case_name)
-    return solver
-
-
-@fluent_launcher_args("3ddp")
-def mixing_elbow_param_case_data_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file("elbow_param.cas.h5", "pyfluent/mixing_elbow")
-    download_file("elbow_param.dat.h5", "pyfluent/mixing_elbow")
-    solver.settings.file.read(file_type="case-data", file_name=case_name)
-    return solver
-
-
-@fluent_launcher_args("2ddp")
-def disk_settings_session(globals):
-    solver = new_solver_session_2d(globals)
-    case_name = download_file("disk.cas.h5", "pyfluent/rotating_disk")
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-@fluent_launcher_args("2ddp")
-def disk_case_session(globals):
-    solver = new_solver_session_2d(globals)
-    case_name = download_file("disk.cas.h5", "pyfluent/rotating_disk")
-    solver.file.read(file_type="case", file_name=case_name)
-    return solver
-
-
-@fluent_launcher_args("3ddp")
-def periodic_rot_settings_session(globals):
-    solver = new_solver_session(globals)
-    case_name = download_file(
-        "periodic_rot.cas.h5",
-        "pyfluent/periodic_rot",
-    )
-    solver.file.read(
-        file_type="case",
-        file_name=case_name,
-        lightweight_setup=True,
-    )
-    return solver
-
-
-monkeypatch = pytest.MonkeyPatch()
-
-
-def disable_datamodel_cache(globals):
-    monkeypatch.setattr(pyfluent, "DATAMODEL_USE_STATE_CACHE", False)
-
-
-def display_names_as_keys_in_cache(globals):
-    DataModelCache.use_display_name = True
-
-
-def new_meshing_session2(globals):
-    session = pyfluent.launch_fluent(mode=pyfluent.LaunchMode.MESHING)
-    return session
-
-
-def new_solver_session2(globals):
-    session = pyfluent.launch_fluent()
-    return session
-
-
-def static_mixer_case_session2(globals):
-    session = new_solver_session2(globals)
-    case_name = download_file("Static_Mixer_main.cas.h5", "pyfluent/static_mixer")
-    session.file.read(file_type="case", file_name=case_name)
-    return session