test_cli

Complexity

Total Complexity

12

Size/Duplication

Total Lines	89
Duplicated Lines	0 %

Importance

Changes

0

7 Functions

Rating	Name	Duplication	Size	Complexity
A	test_save_model_reference()	0	10	2
A	test_save_malformed_model_reference()	0	7	2
A	test_save_model()	0	7	2
A	runner()	0	3	1
A	test_import_from_string__error()	0	8	2
A	test_import_from_string()	0	10	1
A	test_import_from_string__not_str()	0	5	2

import sys
from pathlib import Path

import pytest
from click.testing import CliRunner

import graphinate
from graphinate.cli import ImportFromStringError, cli, import_from_string

EXAMPLES_MATH = 'examples/math'


@pytest.fixture
def runner():
    return CliRunner()


def test_save_model(octagonal_graph_model, runner):
    with runner.isolated_filesystem():
        # Act
        result = runner.invoke(cli, ['save', '-m', octagonal_graph_model])

        # Assert
        assert result.exit_code == 0


def test_save_model_reference(runner):
    # Arrange
    sys.path.append(str(Path(EXAMPLES_MATH).resolve()))

    with runner.isolated_filesystem():
        # Act
        result = runner.invoke(cli, ['save', '-m', "polygonal_graph:model"])

        # Assert
        assert result.exit_code == 0


def test_save_malformed_model_reference(runner):
    with runner.isolated_filesystem():
        # Act
        result = runner.invoke(cli, ['save', '-m', "malformed_model_reference"])

    # Assert
    assert result.exit_code == 2


def test_import_from_string():
    # Arrange
    sys.path.append(EXAMPLES_MATH)

    # Act
    actual = import_from_string("polygonal_graph:model")

    # Assert
    assert isinstance(actual, graphinate.GraphModel)
    assert actual.name == "Octagonal Graph"


import_from_string_error_cases = [
    ("does_not_exist:model", "Could not import module 'does_not_exist'."),
    ("polygonal_graph:does_not_exist",
     "Attribute 'does_not_exist' not found in import string reference 'polygonal_graph:does_not_exist'."),
    ("wrong_format", "Import string 'wrong_format' must be in format '<module>:<attribute>'.")
]


@pytest.mark.parametrize(('case', 'message'), import_from_string_error_cases)
def test_import_from_string__error(case, message):
    # Arrange
    sys.path.append(EXAMPLES_MATH)

    # Act & Assert
    with pytest.raises(ImportFromStringError, match=message):
        _ = import_from_string(case)


import_from_string_not_str_cases = [
    0,
    None
]


@pytest.mark.parametrize('case', import_from_string_not_str_cases)
def test_import_from_string__not_str(case):
    # Act & Assert
    with pytest.raises(ImportFromStringError, match=f"{case} is not a string"):
        _actual = import_from_string(case)