TEST-modin-project#5008: Set benchmark mode within unit test instead of with environment variable (modin-project#6359)

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

Author: anmyachev

.github/workflows/ci.yml

@@ -350,7 +350,7 @@ jobs:
         run: |
           sudo docker pull postgres
           sudo docker run --name some-postgres -e POSTGRES_USER=sa -e POSTGRES_PASSWORD=Strong.Pwd-123 -e POSTGRES_DB=postgres -d -p 2345:5432 postgres
-      - run: MODIN_BENCHMARK_MODE=True mpiexec -n 1 python -m pytest modin/pandas/test/internals/test_benchmark_mode.py
+      - run: mpiexec -n 1 python -m pytest modin/pandas/test/internals/test_benchmark_mode.py
       - run: mpiexec -n 1 python -m pytest modin/pandas/test/internals/test_repartition.py
       - run: mpiexec -n 1 python -m pytest modin/test/test_partition_api.py
       - uses: ./.github/actions/run-core-tests
@@ -447,7 +447,7 @@ jobs:
           sudo docker run --name some-postgres -e POSTGRES_USER=sa -e POSTGRES_PASSWORD=Strong.Pwd-123 -e POSTGRES_DB=postgres -d -p 2345:5432 postgres
         if: matrix.os == 'ubuntu'
 
-      - run: MODIN_BENCHMARK_MODE=True python -m pytest modin/pandas/test/internals/test_benchmark_mode.py
+      - run: python -m pytest modin/pandas/test/internals/test_benchmark_mode.py
         if: matrix.engine == 'python' || matrix.test_task == 'group_1'
       - run: python -m pytest modin/pandas/test/internals/test_repartition.py
         if: matrix.engine == 'python' || matrix.test_task == 'group_1'

modin/conftest.py

@@ -62,6 +62,7 @@ def _saving_make_api_url(token, _make_api_url=modin.utils._make_api_url):
     CIAWSAccessKeyID,
     CIAWSSecretAccessKey,
     AsyncReadMode,
+    BenchmarkMode,
 )
 import uuid  # noqa: E402
 
@@ -558,6 +559,14 @@ def set_num_partitions(request):
     NPartitions.put(old_num_partitions)
 
 
+@pytest.fixture()
+def set_benchmark_mode(request):
+    old_benchmark_mode = BenchmarkMode.get()
+    BenchmarkMode.put(request.param)
+    yield
+    BenchmarkMode.put(old_benchmark_mode)
+
+
 @pytest.fixture
 def set_async_read_mode(request):
     old_async_read_mode = AsyncReadMode.get()

modin/pandas/test/internals/test_benchmark_mode.py

@@ -13,9 +13,10 @@
 
 import unittest.mock as mock
 
+import pytest
+
 import modin.pandas as pd
-from modin.pandas.test.utils import test_data_values
-from modin.config import BenchmarkMode, Engine
+from modin.config import Engine
 
 
 engine = Engine.get()
@@ -46,26 +47,17 @@
     )
 
 
-def test_from_environment_variable():
-    assert BenchmarkMode.get()
-    with mock.patch(wait_method) as wait:
-        pd.DataFrame(test_data_values[0]).mean()
-
-    wait.assert_called()
-
-
-def test_turn_off():
+@pytest.mark.parametrize("set_benchmark_mode", [False], indirect=True)
+def test_turn_off(set_benchmark_mode):
     df = pd.DataFrame([0])
-    BenchmarkMode.put(False)
     with mock.patch(wait_method) as wait:
         df.dropna()
     wait.assert_not_called()
 
 
-def test_turn_on():
-    BenchmarkMode.put(False)
+@pytest.mark.parametrize("set_benchmark_mode", [True], indirect=True)
+def test_turn_on(set_benchmark_mode):
     df = pd.DataFrame([0])
-    BenchmarkMode.put(True)
     with mock.patch(wait_method) as wait:
         df.dropna()
     wait.assert_called()