Spaces:
Running
Running
| import subprocess | |
| import unittest | |
| def run_command(command): | |
| """ | |
| Retrieve output of a command string, decode and convert from CRLF to LF formatting | |
| """ | |
| return subprocess.run(command.split(" "), stdout=subprocess.PIPE).stdout.decode("utf-8").replace("\r\n", "\n") | |
| def make_command(command): | |
| """ | |
| Convert string into list with LF formatting (for unix systems), using \n delimiter | |
| """ | |
| 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) | |
| 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) | |
| def runtests(): | |
| """Run all tests in cliTest.py.""" | |
| loader = unittest.TestLoader() | |
| suite = unittest.TestSuite() | |
| suite.addTests(loader.loadTestsFromTestCase(TestCli)) | |
| runner = unittest.TextTestRunner() | |
| return runner.run(suite) | |