Spaces:
Running
Running
| 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() | |