LU-16459 tests: fix YAML verification function

[fs/lustre-release.git] / lustre / tests / test-framework.sh

diff --git a/lustre/tests/test-framework.sh b/lustre/tests/test-framework.sh
index 673dfc3..478b6c5 100755 (executable)
--- a/lustre/tests/test-framework.sh
+++ b/lustre/tests/test-framework.sh
@@ -11067,9 +11067,9 @@ exhaust_all_precreations() {
 }
 
 verify_yaml_available() {
-       python3 -c "import yaml; yaml.safe_load_all('''a: b''')"
+       python3 -c "import yaml; yaml.safe_load('''a: b''')"
 }
 
 verify_yaml() {
-       python3 -c "import yaml; yaml.safe_load_all('''$1''')"
+       python3 -c "import sys, yaml; obj = yaml.safe_load(sys.stdin)"
 }