Remove need for YAML in unittests

pysr/test/test.py

@@ -10,7 +10,6 @@ import pandas as pd
 import warnings
 import pickle as pkl
 import tempfile
-import yaml
 from pathlib import Path
 
 from .. import julia_helpers
@@ -718,29 +717,22 @@ class TestMiscellaneous(unittest.TestCase):
         param_groupings_file = (
             Path(__file__).parent.parent.parent / "docs" / "param_groupings.yml"
         )
-        # Read the file:
+        # Read the file, discarding lines ending in ":",
+        # and removing leading "\s*-\s*":
+        params = []
         with open(param_groupings_file, "r") as f:
-            param_groupings = yaml.load(f, Loader=yaml.SafeLoader)
-
-        # Get all leafs of this yaml file:
-        def get_leafs(d):
-            if isinstance(d, dict):
-                for v in d.values():
-                    yield from get_leafs(v)
-            elif isinstance(d, list):
-                for v in d:
-                    yield from get_leafs(v)
-            else:
-                yield d
-
-        leafs = list(get_leafs(param_groupings))
+            for line in f.readlines():
+                if line.strip().endswith(":"):
+                    continue
+                if line.strip().startswith("-"):
+                    params.append(line.strip()[1:].strip())
 
         regressor_params = [
             p for p in DEFAULT_PARAMS.keys() if p not in ["self", "kwargs"]
         ]
 
         # Check the sets are equal:
-        self.assertSetEqual(set(leafs), set(regressor_params))
+        self.assertSetEqual(set(params), set(regressor_params))
 
 
 TRUE_PREAMBLE = "\n".join(