Extended the API to find an unique Yaml node.

1 files changed, 6 insertions, 6 deletions

diff --git a/tests/plugins/yamlrdr.py b/tests/plugins/yamlrdr.py
index eeae032..2f2694b 100644
--- a/tests/plugins/yamlrdr.py
+++ b/tests/plugins/yamlrdr.py
@@ -242,10 +242,10 @@ root:
 
         if len(found) == 2:
 
-            sub = found[0].find_by_path('/a')
-            self.assertEqual(len(sub), 1)
-            self.assertEqual(sub[0].yaml_line.key, 'a')
+            sub = found[0].find_one_by_path('/a')
+            self.assertIsNotNone(sub)
+            self.assertEqual(sub.yaml_line.key, 'a')
 
-            sub = found[0].find_by_path('/aa')
-            self.assertEqual(len(sub), 1)
-            self.assertEqual(sub[0].yaml_line.key, 'aa')
+            sub = found[0].find_one_by_path('/aa')
+            self.assertIsNotNone(sub)
+            self.assertEqual(sub.yaml_line.key, 'aa')