test_modeling

Complexity

Total Complexity

31

Size/Duplication

Total Lines	198
Duplicated Lines	0 %

Importance

Changes

0

21 Functions

Rating	Name	Duplication	Size	Complexity
A	test_validate_type_raises_on_invalid_identifier()	0	4	2
A	test_elements_empty_iterable()	0	4	1
A	test_elements_callable_element_type()	0	14	1
A	test_validate_type_raises_on_empty_string()	0	4	2
A	test_validate_type_accepts_callable()	0	6	1
A	test_elements_invalid_identifier_type()	0	8	2
A	test_elements_multiple_getters()	0	11	2
A	test_validate_type_non_callable_valid_identifier()	0	4	1
A	test_validate_type_accepts_identifier_and_callable()	0	8	1
A	test_validate_type_accepts_valid_identifier()	0	4	1
A	test_graph_model_validate_node_parameters()	0	7	2
A	test_validate_type_non_string_callable()	0	8	1
A	test_graph_model__add__()	0	7	1
A	test_validate_type_empty_string()	0	4	2
A	test_validate_type_neither_callable_nor_identifier()	0	4	2
A	test_elements_yields_correct_count()	0	11	2
A	test_graph_model()	0	15	1
A	test_validate_type_special_characters()	0	4	2
A	test_elements_element_type_not_callable_invalid()	0	5	2
A	test_validate_type_callable_string()	0	10	1
A	test_validate_type_identifier_and_callable()	0	8	1

import pytest

import graphinate
import graphinate.typing
from graphinate.modeling import GraphModel, elements


def test_graph_model(map_graph_model):
    # arrange
    expected_country_count, expected_city_count, graph_model = map_graph_model
    country_type_id = (graphinate.typing.UniverseNode, 'country')
    city_type_id = ('country', 'city')

    # act
    actual_model_count = len(graph_model._node_models)
    actual_country_count = len(list(graph_model._node_models[country_type_id][0].generator()))
    actual_city_count = len(list(graph_model._node_models[city_type_id][0].generator()))

    # assert
    assert actual_model_count == 3
    assert actual_country_count == expected_country_count  # len(country_ids)
    assert actual_city_count == expected_city_count  # len(city_ids)


def test_graph_model__add__():
    first_model = graphinate.model(name='First Model')
    second_model = graphinate.model(name='Second Model')

    actual_model = first_model + second_model

    assert actual_model.name == 'First Model + Second Model'


def test_graph_model_validate_node_parameters():
    graph_model = graphinate.model(name='Graph with invalid node supplier')

    with pytest.raises(graphinate.modeling.GraphModelError):
        @graph_model.node()
        def invalid_node_supplier(wrong_parameter=None):
            yield 1


def test_validate_type_identifier_and_callable():
    class CallableStr(str):
        def __call__(self):
            return "called"

    node_type = CallableStr("valididentifier")
    # Should not raise
    GraphModel._validate_type(node_type)


def test_validate_type_non_callable_valid_identifier():
    node_type = "valid_identifier"
    # Should not raise
    GraphModel._validate_type(node_type)


def test_validate_type_non_string_callable():
    class DummyCallable:
        def __call__(self):
            return "called"

    node_type = DummyCallable()
    # Should not raise
    GraphModel._validate_type(node_type)


def test_validate_type_neither_callable_nor_identifier():
    node_type = "123 invalid!"
    with pytest.raises(ValueError, match="Invalid Type:"):
        GraphModel._validate_type(node_type)


def test_validate_type_empty_string():
    node_type = ""
    with pytest.raises(ValueError, match="Invalid Type:"):
        GraphModel._validate_type(node_type)


def test_validate_type_special_characters():
    node_type = "invalid-type!"
    with pytest.raises(ValueError, match="Invalid Type:"):
        GraphModel._validate_type(node_type)


def test_validate_type_accepts_valid_identifier():
    node_type = "node_type"
    # Should not raise
    GraphModel._validate_type(node_type)


def test_validate_type_accepts_callable():
    def some_callable():
        pass

    # Should not raise
    GraphModel._validate_type(some_callable)


def test_validate_type_accepts_identifier_and_callable():
    class CallableStr(str):
        def __call__(self):
            return "called"

    node_type = CallableStr("anotheridentifier")
    # Should not raise
    GraphModel._validate_type(node_type)


def test_validate_type_raises_on_invalid_identifier():
    node_type = "not valid!"
    with pytest.raises(ValueError, match="Invalid Type:"):
        GraphModel._validate_type(node_type)


def test_validate_type_raises_on_empty_string():
    node_type = ""
    with pytest.raises(ValueError, match="Invalid Type:"):
        GraphModel._validate_type(node_type)


def test_validate_type_callable_string():
    # In Python, a string is never callable, but for the sake of the test, let's simulate
    # a string subclass that is callable
    class CallableStr(str):
        def __call__(self):
            return "called"

    node_type = CallableStr("not_an_identifier_but_callable")
    # Should not raise
    GraphModel._validate_type(node_type)


def test_elements_yields_correct_count():
    data = [
        {"id": 1, "name": "A"},
        {"id": 2, "name": "B"},
        {"id": 3, "name": "C"},
    ]
    result = list(elements(data, element_type="Node", id="id", name="name"))
    assert len(result) == len(data)
    for i, el in enumerate(result):
        assert el.id == data[i]["id"]
        assert el.name == data[i]["name"]


def test_elements_multiple_getters():
    data = [
        {"id": 1, "name": "A", "value": 10},
        {"id": 2, "name": "B", "value": 20},
    ]
    result = list(elements(data, element_type="Node", id="id", name="name", value="value"))
    assert len(result) == 2
    for i, el in enumerate(result):
        assert el.id == data[i]["id"]
        assert el.name == data[i]["name"]
        assert el.value == data[i]["value"]


def test_elements_empty_iterable():
    data = []
    result = list(elements(data, element_type="Node", id="id"))
    assert result == []


def test_elements_element_type_not_callable_invalid():
    data = [{"id": 1}]
    # element_type is not callable and not a valid identifier
    with pytest.raises(ValueError, match="Invalid Type:"):
        list(elements(data, element_type="123 invalid!", id="id"))


def test_elements_callable_element_type():
    data = [
        {"type": "Alpha", "id": 1},
        {"type": "Beta", "id": 2},
    ]

    def type_extractor(item):
        return item["type"]

    result = list(elements(data, element_type=type_extractor, id="id"))
    assert result[0].__class__.__name__ == "Alpha"
    assert result[1].__class__.__name__ == "Beta"
    assert result[0].id == 1
    assert result[1].id == 2


def test_elements_invalid_identifier_type():
    data = [{"id": 1}]

    def bad_type(item):
        return "not valid!"

    with pytest.raises(ValueError, match="Invalid Type:"):
        list(elements(data, element_type=bad_type, id="id"))