test_classification_rule_data

Tests for ClassificationRuleData class

test_classification_rule_data_creation_logic()

The ClassificationRuleData should parse the provided dictionary to correctly initialize itself during creation

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

def test_classification_rule_data_creation_logic():
    """The ClassificationRuleData should parse the provided dictionary
    to correctly initialize itself during creation"""

    # Arrange
    test_table = {"a": [1], "output": [2]}

    # Act
    data = ClassificationRuleData("test_name", ["foo", "bar"], test_table)
    data.description = "description"

    # Assert

    assert isinstance(data, IRuleData)
    assert data.criteria_table == test_table
    assert data.input_variable_names == ["foo", "bar"]