test_examples

Complexity

Total Complexity

4

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	TestTutorial.test_registry()	0	11	1

import subprocess

import pytest

from deepreg import log

logger = log.get(__name__)


def execute_commands(cmds):
    for cmd in cmds:
        try:
            logger.info(f"Running {cmd}")
            out = subprocess.check_output(cmd, shell=True).decode("utf-8")
            logger.info(out)
        except subprocess.CalledProcessError as e:
            raise RuntimeError(
                f"Command {cmd} return with err {e.returncode} {e.output}"
            )


class TestTutorial:
    @pytest.mark.parametrize(
        "name",
        [
            "custom_backbone",
            "custom_image_label_loss",
            "custom_parameterized_image_label_loss",
        ],
    )
    def test_registry(self, name: str):
        cmds = [f"python examples/{name}.py"]
        execute_commands(cmds)