leftyer/mindformers
1
0
Fork 0
mirror of https://gitee.com/mindspore/mindformers.git synced 2025-12-06 11:29:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

!6744 【dev】【用例】【bugfix】修复yaml深度测试用例问题

Browse Source 
Merge pull request !6744 from zyw_hw/fix_check_rules_case
This commit is contained in:
i-robot
2025-07-14 02:55:16 +00:00
committed by Gitee
parent e0e02f5e3d cbbef42a83
commit a03af14a96
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/st/test_ut/test_check_rules.py
View File

@@ -53,15 +53,15 @@ class TestCheckYamlDepthBeforeLoading(unittest.TestCase):
assert check_yaml_depth_before_loading(self.yaml_str)
with open(self.yaml_path, encoding='utf-8') as fp:
node = yaml.compose(fp)
assert get_yaml_ast_depth(node) == 3
assert get_yaml_ast_depth(node) == 4
fp.seek(0)
check_yaml_depth_before_loading(fp)
fp.seek(0)
cfg_dict = yaml.safe_load(fp)
assert cfg_dict["model"]["model_config"]["num_layers"] == 32
assert cfg_dict["model"]["model_config"]["num_layers"] == 40
fp.seek(0)
cfg_dict = yaml.load(fp, Loader=yaml.SafeLoader)
assert cfg_dict["model"]["model_config"]["num_layers"] == 32
assert cfg_dict["model"]["model_config"]["num_layers"] == 40
fp.seek(0)
cfg_dict = yaml.safe_load(fp.read())
assert cfg_dict["model"]["model_config"]["num_layers"] == 32
assert cfg_dict["model"]["model_config"]["num_layers"] == 40