Spaces:
Sleeping
Sleeping
File size: 2,373 Bytes
8e9aba4 976f8d8 4726b4a 976f8d8 3555cfd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import unittest
from click import testing as click_testing
def get_runtests():
# Lazy load to avoid circular imports.
from .._cli.main import pysr
class TestCli(unittest.TestCase):
# TODO: Include test for custom project here.
def setUp(self):
self.cli_runner = click_testing.CliRunner()
def test_help_on_all_commands(self):
expected = "\n".join(
[
"Usage: pysr [OPTIONS] COMMAND [ARGS]...",
"",
"Options:",
" --help Show this message and exit.",
"",
"Commands:",
" install Install Julia dependencies for PySR.",
"",
]
)
result = self.cli_runner.invoke(pysr, ["--help"])
self.assertEqual(expected, result.output)
self.assertEqual(0, result.exit_code)
def test_help_on_install(self):
expected = "\n".join(
[
"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.",
"",
]
)
result = self.cli_runner.invoke(pysr, ["install", "--help"])
self.assertEqual(expected, result.output)
self.assertEqual(0, result.exit_code)
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)
return runtests
|