test_yaml_model_data

Tests for YamlModelData class

test_yaml_model_data_default_settings_and_type()

Test if the YamlModelData implements the IModelData interface and gives the right default settings

Source code in tests/data/entities/test_yaml_model_data.py

def test_yaml_model_data_default_settings_and_type():
    """Test if the YamlModelData implements the IModelData
    interface and gives the right default settings"""

    # Arrange
    datasets = [Mock(DatasetData)]
    rules = [Mock(MultiplyRuleData)]
    version = [0, 0, 0]

    # Act
    model_data = YamlModelData("Model 1", version)
    model_data.output_path = Path("")
    model_data.datasets = datasets
    model_data.output_variables = []
    model_data.rules = rules

    # Assert

    # implements interface
    assert isinstance(model_data, IModelData)

    assert model_data.name == "Model 1"
    assert model_data.datasets == datasets
    assert isinstance(model_data.datasets[0], IDatasetData)
    assert model_data.rules == rules
    assert isinstance(model_data.rules[0], IRuleData)