def test_list(task):
"""Test function listing."""
result = np.array(
main(
["list", "--task",
task.__name__.split(".")[-1], "--datasets"],
do_print=False,
))
expected = np.array([dataset.__name__ for dataset in task.DATASETS])
assert np.all(result == expected)
result = np.array(
main(
["list", "--task",
task.__name__.split(".")[-1], "--methods"],
do_print=False,
))
expected = np.array([method.__name__ for method in task.METHODS])
assert np.all(result == expected)
result = np.array(
main(
["list", "--task",
task.__name__.split(".")[-1], "--metrics"],
do_print=False,
))
expected = np.array([metric.__name__ for metric in task.METRICS])
assert np.all(result == expected)
def test_hash(task, function_type, function_name):
"""Test git hash function."""
h1 = main(
["hash", "--task", task, function_type, function_name],
do_print=False,
)
h2 = main(
["hash", "--task", task, function_type, function_name],
do_print=False,
)
assert h1 == h2
def test_tasks(capsys):
"""Test task listing."""
result = np.array(main(["tasks"], do_print=False))
expected = np.array(
[task.__name__.split(".")[-1] for task in openproblems.TASKS])
assert np.all(result == expected)
result = np.array(main(["tasks"], do_print=True))
expected = ("\n".join(
[task.__name__.split(".")[-1] for task in openproblems.TASKS]) + "\n")
captured = capsys.readouterr()
assert captured.out == expected
def test_zero_metric():
def __zero_metric(*args):
return 0.0
metric_name = utils.name.object_name(__zero_metric)
task = openproblems.TASKS[0]
setattr(task.metrics, metric_name, __zero_metric)
adata = task.api.sample_dataset()
with tempfile.TemporaryDirectory() as tempdir:
dataset_file = os.path.join(tempdir, "dataset.h5ad")
adata.write_h5ad(dataset_file)
result = main(
[
"evaluate",
"--task",
task.__name__.split(".")[-1],
"--input",
dataset_file,
metric_name,
],
do_print=True,
)
assert result == 0
assert isinstance(result, int)
def test_help(capsys):
assert main([], do_print=False) is None
captured = capsys.readouterr()
assert len(captured.out) > 0
assert (
"Open Problems for Single Cell Analysis command-line interface" in captured.out
)
def test_pipeline():
"""Test evaluation pipeline."""
with tempfile.TemporaryDirectory() as tempdir:
dataset_file = os.path.join(tempdir, "dataset.h5ad")
method_file = os.path.join(tempdir, "method.h5ad")
assert not os.path.isfile(dataset_file)
assert not os.path.isfile(method_file)
main(
[
"load",
"--task",
"label_projection",
"--test",
"--output",
dataset_file,
"pancreas_batch",
],
do_print=False,
)
assert os.path.isfile(dataset_file)
main(
[
"run",
"--task",
"label_projection",
"--input",
dataset_file,
"--output",
method_file,
"logistic_regression_log_cpm",
],
do_print=False,
)
assert os.path.isfile(method_file)
result = main(
[
"evaluate",
"--task",
"label_projection",
"--input",
method_file,
"accuracy",
],
do_print=False,
)
assert isinstance(result, float)
def _test_image(task, function_type, function):
result = main(
[
"image",
"--task",
task.__name__.split(".")[-1],
function_type,
function.__name__,
],
do_print=False,
)
expected = function.metadata["image"]
assert result == expected
def test_version():
assert main(["--version"], do_print=False) == openproblems.__version__
def test_hash_basic():
assert main(["--test-hash"], do_print=False) is None
