test_classification_rule

Tests for Classification class

test_create_classification_rule_should_set_defaults()

Test creating a classification rule with defaults

Source code in tests/business/entities/rules/test_classification_rule.py

def test_create_classification_rule_should_set_defaults():
    """Test creating a classification rule with defaults"""

    # test data
    criteria_test_table = {
        "output": [1, 2, 3, 4],
        "water_depth": [0.1, 3.33, 5, 5],
        "temperature": ["-", "0.1: 15", 15, ">15"],
    }

    # Arrange and act
    rule = ClassificationRule("test", ["water_depth", "salinity"], criteria_test_table)

    # assert
    assert rule.name == "test"
    assert rule.input_variable_names == ["water_depth", "salinity"]
    assert rule.criteria_table == criteria_test_table
    assert rule.output_variable_name == "output"
    assert rule.description == ""
    assert isinstance(rule, ClassificationRule)

test_execute_classification()

Test executing a classification of values

Source code in tests/business/entities/rules/test_classification_rule.py

def test_execute_classification():
    """Test executing a classification of values"""

    # test data
    criteria_test_table = {
        "output": [100, 200, 300, 400, 500, 900, 111, 222, 333],
        "water_depth": [11, 12, 13, 13, 15, 0, "-", "0", "0"],
        "salinity": ["-", "0.5: 5.5", 8.8, 8.8, 9, 0, ">10", "0", "0"],
        "temperature": ["-", "-", "-", "-", ">25.0", 0, "<0", "0", "0"],
        "another_val": ["-", "-", "-", "-", "-", "<0", ">0", ">=33", "<=24"],
    }

    # arrange
    logger = Mock(ILogger)
    rule = ClassificationRule("test", ["water_depth", "salinity"], criteria_test_table)
    test_data = {
        "water_depth": _xr.DataArray([13, 0, 11, 15, 12, 20, 0, 0]),
        "salinity": _xr.DataArray([8.8, 0, 2, 9, 2.5, 11, 0, 0]),
        "temperature": _xr.DataArray([20, -5, 20, 28, 1, -5, 0, 0]),
        "another_val": _xr.DataArray([1, 2, 3, 4, 5, 9, 22, 33]),
    }

    # expected results:
    # 1: take first when multiple apply --> 300
    # 2: no possible classification --> None
    # 3: allow '-' --> 100
    # 4: greater than '>' --> 500
    # 5: range --> 200
    # 6: smaller than '<' --> 111
    # 7: greater than/equal to '>=' --> 222
    # 8: smaller than/equal to '<=' --> 333
    expected_result = _xr.DataArray([300, None, 100, 500, 200, 111, 333, 222])

    # act
    test_result = rule.execute(test_data, logger)

    # assert
    assert _xr.testing.assert_equal(test_result, expected_result) is None