Skip SQL count check in test (#1721)

tests/integ/modin/conftest.py

@@ -45,6 +45,7 @@ def pytest_addoption(parser):
     parser.addoption(
         "--generate_pandas_api_coverage", action="store_true", default=False
     )
+    parser.addoption("--skip_sql_count_check", action="store_true", default=False)
 
 
 @pytest.fixture(scope="session", autouse=True)
@@ -54,6 +55,17 @@ def setup_pandas_api_coverage_generator(pytestconfig):
         PandasAPICoverageGenerator()
 
 
+SKIP_SQL_COUNT_CHECK = False
+
+
+@pytest.fixture(scope="session", autouse=True)
+def setup_skip_sql_count_check(pytestconfig):
+    skip = pytestconfig.getoption("skip_sql_count_check")
+    if skip:
+        global SKIP_SQL_COUNT_CHECK
+        SKIP_SQL_COUNT_CHECK = True
+
+
 @pytest.fixture(scope="function")
 def sql_counter():
     """Return a sql counter as pytest fixture"""

tests/integ/modin/sql_counter.py

@@ -119,8 +119,14 @@ def __init__(
         high_count_reason=None,
         **kwargs,
     ) -> "SqlCounter":
+        from tests.integ.modin.conftest import SKIP_SQL_COUNT_CHECK
+
         self._queries: list[QueryRecord] = []
-        self._no_check = no_check
+
+        # Bypassing sql counter since
+        #   1. it is an unnecessary metric for tests running in stored procedures
+        #   2. pytest-assume package is not available in conda
+        self._no_check = no_check or IS_IN_STORED_PROC or SKIP_SQL_COUNT_CHECK
 
         # Save any expected sql counts initialized at start up.
         self._expected_sql_counts = {}
@@ -357,12 +363,6 @@ def sql_count_checker(
     *args,
     **kwargs,
 ):
-    # Bypassing sql counter since
-    #   1. it is an unnecessary metric for tests running in stored procedures
-    #   2. pytest-assume package is not available in conda
-    if IS_IN_STORED_PROC:
-        return
-
     """SqlCounter decorator that automatically validates the sql counts when test finishes."""
     sql_counter = SqlCounter(
         no_check=no_check,