Added tests to check help messages

pysr/test/cliTest.py

@@ -0,0 +1,54 @@
+import io
+import subprocess
+import sys
+import unittest
+
+
+def run_command(command):
+    return subprocess.run(command.split(" "), stdout=subprocess.PIPE).stdout.decode("utf-8").replace("\r\n", "\n")
+
+
+def make_command(command):
+    return "\n".join(command)
+
+
+class TestCli(unittest.TestCase):
+
+    def test_help_on_all_commands(self):
+        command_to_test = "python -m pysr --help"
+        expected_lines = ["Usage: pysr [OPTIONS] COMMAND [ARGS]...",
+                          "",
+                          "Options:",
+                          "  --help  Show this message and exit.",
+                          "",
+                          "Commands:",
+                          "  install  Install Julia dependencies for PySR.",
+                          ""]
+
+        expected = make_command(expected_lines)
+        actual = run_command(command_to_test)
+        self.assertEqual(expected, actual)  # add assertion here
+
+    def test_help_on_install(self):
+        command_to_test = "python -m pysr install --help"
+        expected_lines = ["Usage: pysr install [OPTIONS]",
+                          "",
+                          "  Install Julia dependencies for PySR.",
+                          "",
+                          "Options:",
+                          "  -p, --project PROJECT_DIRECTORY",
+                          "                                  Install in a specific Julia project (e.g., a",
+                          "                                  local copy of SymbolicRegression.jl).",
+                          "  -q, --quiet                     Disable logging.",
+                          "  --precompile                    Force precompilation of Julia libraries.",
+                          "  --no-precompile                 Disable precompilation.",
+                          "  --help                          Show this message and exit.",
+                          ""]
+
+        expected = make_command(expected_lines)
+        actual = run_command(command_to_test)
+        self.assertEqual(expected, actual)
+
+
+if __name__ == '__main__':
+    unittest.main()