diff --git a/aeneas/tests/common.py b/aeneas/tests/common.py index dac7e16..4cb309e 100644 --- a/aeneas/tests/common.py +++ b/aeneas/tests/common.py @@ -13,12 +13,19 @@ (val := os.getenv("UNITTEST_RUN_SLOW_TESTS")) is None or val.strip() == "0", "slow tests are disabled. Set `UNITTEST_RUN_SLOW_TESTS=1` in the environment to enable them.", ) + EXTRA_TEST_PATH = os.path.join(os.path.expanduser("~"), ".aeneas.conf") extra_test = unittest.skipIf( not os.path.isfile(EXTRA_TEST_PATH), f"extra tests are disabled (path {EXTRA_TEST_PATH!r} does not exist).", ) +BENCH_DIR = os.path.join(os.path.expanduser("~"), ".aeneas", "benchmark_input") +bench_test = unittest.skipIf( + not os.path.isdir(BENCH_DIR), + f"bench tests are disabled (directory {BENCH_DIR!r} does not exist).", +) + class ExecuteCLICase(unittest.TestCase): CLI_CLS: typing.ClassVar diff --git a/aeneas/tests/bench_test_task.py b/aeneas/tests/test_bench_task.py similarity index 98% rename from aeneas/tests/bench_test_task.py rename to aeneas/tests/test_bench_task.py index 80df07d..dcb02a5 100644 --- a/aeneas/tests/bench_test_task.py +++ b/aeneas/tests/test_bench_task.py @@ -26,16 +26,12 @@ import tempfile from aeneas.tools.execute_task import ExecuteTaskCLI +from aeneas.tests.common import BENCH_DIR, bench_test -BENCH_DIR = os.path.join(os.path.expanduser("~"), ".aeneas", "benchmark_input") -BENCH_TESTS = os.path.exists(BENCH_DIR) - - +@bench_test class TestBenchmarkExecuteTaskCLI(unittest.TestCase): def bench_execute(self, parameters, expected_exit_code, timeout): - if not BENCH_TESTS: - return args = (parameters, expected_exit_code) p = multiprocessing.Process(target=self.execute, name="funcExecute", args=args) p.start() diff --git a/aeneas/tests/bench_test_task_misc.py b/aeneas/tests/test_bench_task_misc.py similarity index 98% rename from aeneas/tests/bench_test_task_misc.py rename to aeneas/tests/test_bench_task_misc.py index da87c44..8df49bd 100644 --- a/aeneas/tests/bench_test_task_misc.py +++ b/aeneas/tests/test_bench_task_misc.py @@ -26,16 +26,12 @@ import tempfile from aeneas.tools.execute_task import ExecuteTaskCLI +from aeneas.tests.common import BENCH_DIR, bench_test -BENCH_DIR = os.path.join(os.path.expanduser("~"), ".aeneas", "benchmark_input") -BENCH_TESTS = os.path.exists(BENCH_DIR) - - +@bench_test class TestBenchmarkExecuteTaskCLI(unittest.TestCase): def bench_execute(self, parameters, expected_exit_code, timeout): - if not BENCH_TESTS: - return args = (parameters, expected_exit_code) p = multiprocessing.Process(target=self.execute, name="funcExecute", args=args) p.start()