Files
opm-simulators/python/test/test_mpi.py
T
Håkon Hægland c195df978d Refactor: Use factory fixtures for simulator creation in tests
Move simulator instantiation to factory functions that automatically
disable async ECL output. This prevents future contributors from
forgetting the --enable-async-ecl-output=false flag.

- Add create_black_oil_simulator/create_gas_water_simulator to pytest_common
- Define ENABLE_ASYNC_ECL_OUTPUT_FLAG constant
- Update all tests to use factory functions
- No functional changes, pure refactoring
2025-09-23 05:09:20 +02:00

33 lines
1.4 KiB
Python

import os
import unittest
from pathlib import Path
from .pytest_common import pushd, create_black_oil_simulator
class TestBasic(unittest.TestCase):
@classmethod
def setUpClass(cls):
test_dir = Path(os.path.dirname(__file__))
cls.data_dir = test_dir.parent.joinpath("test_data/SPE1CASE1a")
# IMPORTANT: Since all the python unittests run in the same process we must be
# careful to not call MPI_Init() more than once.
# Tests are run alphabetically, so we need to make sure that
# the the first test calls MPI_Init(), therefore the name of the tests
# have a numeric label like "01" in test_01_mpi_init to ensure that they
# are run in a given order.
def test_01_mpi_init(self):
with pushd(self.data_dir):
sim = create_black_oil_simulator(filename="SPE1CASE1.DATA")
sim.setup_mpi(init=True, finalize=False)
sim.step_init() # This will create the OPM::Main() object which will call MPI_Init()
assert True
def test_02_mpi_no_init(self):
with pushd(self.data_dir):
sim = create_black_oil_simulator(filename="SPE1CASE1.DATA")
sim.setup_mpi(init=False, finalize=True)
sim.step_init() # This will create the OPM::Main() object which will not call MPI_Init()
# That this test runs shows that the simulator does not call
# MPI_Init() a second time
assert True