summaryrefslogtreecommitdiffstats
path: root/yql/essentials/tests/common/test_framework/yql_utils.py
diff options
context:
space:
mode:
authoraneporada <[email protected]>2025-03-10 14:57:14 +0300
committeraneporada <[email protected]>2025-03-10 15:11:53 +0300
commit170ddbacc61d815df8eee981deb1316a689b8e3b (patch)
tree177fec16a466e5f19c575dad3318c3f2e7185e63 /yql/essentials/tests/common/test_framework/yql_utils.py
parent4fb6424a2c472a1d73cf6ca59f73021e06e8484a (diff)
Introduce yt.RuntimeCluster/Selection + test infrastructure changes
yt.RuntimeClusterSelection and yt.RuntimeClster fix Add default yt cluster for yt_file tests Revert "Add plato + banach to yt_file/yt configs" This reverts commit 83b85e67d753abed32fc620991b0ddd7ea58dcee. Add plato + banach to yt_file/yt configs add MrCluster setting Support yt.MrCluster commit_hash:e046a70ebea88ca0adcd497ce2e66d75a3c339ff
Diffstat (limited to 'yql/essentials/tests/common/test_framework/yql_utils.py')
-rw-r--r--yql/essentials/tests/common/test_framework/yql_utils.py15
1 files changed, 13 insertions, 2 deletions
diff --git a/yql/essentials/tests/common/test_framework/yql_utils.py b/yql/essentials/tests/common/test_framework/yql_utils.py
index 2136729be9b..3e4a4afa3fe 100644
--- a/yql/essentials/tests/common/test_framework/yql_utils.py
+++ b/yql/essentials/tests/common/test_framework/yql_utils.py
@@ -151,13 +151,15 @@ Table = namedtuple('Table', (
'yqlrun_file',
'attr',
'format',
- 'exists'
+ 'exists',
+ 'cluster'
))
def new_table(full_name, file_path=None, yqlrun_file=None, content=None, res_dir=None,
attr=None, format_name='yson', def_attr=None, should_exist=False, src_file_alternative=None):
assert '.' in full_name, 'expected name like cedar.Input'
+ cluster = full_name.split('.')[0]
name = '.'.join(full_name.split('.')[1:])
if res_dir is None:
@@ -231,7 +233,8 @@ def new_table(full_name, file_path=None, yqlrun_file=None, content=None, res_dir
new_yqlrun_file,
attr,
format_name,
- exists
+ exists,
+ cluster
)
@@ -463,6 +466,14 @@ def get_tables(suite, cfg, data_path, def_attr=None):
return in_tables, out_tables
+def get_table_clusters(suite, cfg, data_path):
+ in_tables, out_tables = get_tables(suite, cfg, data_path)
+ clusters = set()
+ for t in in_tables + out_tables:
+ clusters.add(t.cluster)
+ return clusters
+
+
def get_supported_providers(cfg):
providers = 'yt', 'kikimr', 'dq', 'hybrid'
for item in cfg: