switch off balance check in perf tests (#11560)

Author: iddqdex

ydb/tests/olap/lib/ydb_cluster.py

@@ -235,27 +235,28 @@ def _check_node(n):
             elif isinstance(balanced_paths, list):
                 for path in balanced_paths:
                     paths_to_balance += cls._get_tables(path)
-            for p in paths_to_balance:
-                table_nodes, _ = cls.get_cluster_nodes(p)
-                min = None
-                max = None
-                for tn in table_nodes:
-                    tablet_count = 0
-                    for tablet in tn.get("Tablets", []):
-                        if tablet.get("State") != "Green":
-                            errors.append(f'Node {tn.get("SystemState", {}).get("Host")}: {tablet.get("Count")} tablets of type {tablet.get("Type")} in {tablet.get("State")} state')
-                        if tablet.get("Type") in {"ColumnShard", "DataShard"}:
-                            tablet_count += tablet.get("Count")
-                    if tablet_count > 0:
-                        if min is None or tablet_count < min:
-                            min = tablet_count
-                        if max is None or tablet_count > max:
-                            max = tablet_count
-                if min is None or max is None:
-                    errors.append(f'Table {p} has no tablets')
-                elif max - min > 1:
-                    errors.append(f'Table {p} is not balanced: {min}-{max} shards.')
-                LOGGER.info(f'Table {p} is balanced: {min}-{max} shards.')
+            if os.getenv('TEST_CHECK_BALANCING', 'no') == 'yes':
+                for p in paths_to_balance:
+                    table_nodes, _ = cls.get_cluster_nodes(p)
+                    min = None
+                    max = None
+                    for tn in table_nodes:
+                        tablet_count = 0
+                        for tablet in tn.get("Tablets", []):
+                            if tablet.get("State") != "Green":
+                                errors.append(f'Node {tn.get("SystemState", {}).get("Host")}: {tablet.get("Count")} tablets of type {tablet.get("Type")} in {tablet.get("State")} state')
+                            if tablet.get("Type") in {"ColumnShard", "DataShard"}:
+                                tablet_count += tablet.get("Count")
+                        if tablet_count > 0:
+                            if min is None or tablet_count < min:
+                                min = tablet_count
+                            if max is None or tablet_count > max:
+                                max = tablet_count
+                    if min is None or max is None:
+                        errors.append(f'Table {p} has no tablets')
+                    elif max - min > 1:
+                        errors.append(f'Table {p} is not balanced: {min}-{max} shards.')
+                    LOGGER.info(f'Table {p} is balanced: {min}-{max} shards.')
 
             cls.execute_single_result_query("select 1", timeout)
         except BaseException as ex: