test_parser_multiply_rule

Tests for ParserMultiplyRule class

test_multiply_parser_with_multipliers_incorrect_headers(multipliers_table, expected_message)

Test when multipliers table is available

Source code in tests/data/parsers/test_parser_multiply_rule.py

@pytest.mark.parametrize(
    "multipliers_table, expected_message",
    [
        (
            [["date", "end_date", "multipliers"], ["01-01", "15-07", [1, 100]]],
            "Missing element start_date",
        ),
        (
            [
                ["start_date", "not_end_date", "multipliers"],
                ["01-01", "15-07", [1, 100]],
            ],
            "Missing element end_date",
        ),
        (
            [
                ["start_date", "end_date", "something_else"],
                ["01-01", "15-07", [1, 100]],
            ],
            "Missing element multipliers",
        ),
    ],
)
def test_multiply_parser_with_multipliers_incorrect_headers(
    multipliers_table: List[List[Any]], expected_message: str
):
    """Test when multipliers table is available"""
    # Arrange
    contents = {
        "name": "testname",
        "input_variable": "input",
        "output_variable": "output",
        "multipliers_table": multipliers_table,
    }
    logger = Mock(ILogger)

    # Act
    data = ParserMultiplyRule()
    with pytest.raises(AttributeError) as exc_info:
        data.parse_dict(contents, logger)

    exception_raised = exc_info.value

    # Assert.
    assert exception_raised.args[0] == expected_message

test_multiply_parser_with_multipliers_table_correct()

Test when multipliers table is available

Source code in tests/data/parsers/test_parser_multiply_rule.py

def test_multiply_parser_with_multipliers_table_correct():
    """Test when multipliers table is available"""
    # Arrange
    contents = {
        "name": "testname",
        "input_variable": "input",
        "output_variable": "output",
        "multipliers_table": [
            ["start_date", "end_date", "multipliers"],
            ["01-01", "15-07", [1, 100]],
            ["16-07", "31-12", [0]],
            ["16-7", "31-12", [1]],
            ["1-11", "31-12", [0]],
        ],
    }
    logger = Mock(ILogger)

    # Act
    data = ParserMultiplyRule()
    parsed_dict = data.parse_dict(contents, logger)

    assert isinstance(parsed_dict, IRuleData)

test_parse_dict_to_rule_data_logic()

Test if a correct dictionary is parsed into a RuleData object

Source code in tests/data/parsers/test_parser_multiply_rule.py

def test_parse_dict_to_rule_data_logic():
    """Test if a correct dictionary is parsed into a RuleData object"""
    # Arrange
    contents = {
        "name": "testname",
        "input_variable": "input",
        "multipliers": [0.0, 1.0],
        "output_variable": "output",
    }
    logger = Mock(ILogger)
    # Act
    data = ParserMultiplyRule()
    parsed_dict = data.parse_dict(contents, logger)

    assert isinstance(parsed_dict, IRuleData)

test_parse_multipliers_type()

Test if an incorrect dictionary is not parsed

Source code in tests/data/parsers/test_parser_multiply_rule.py

def test_parse_multipliers_type():
    """Test if an incorrect dictionary is not parsed"""
    # Arrange
    contents = {
        "name": "testname",
        "input_variable": "input",
        "multipliers": ["a", "b", 2],
        "output_variable": "output",
    }
    logger = Mock(ILogger)

    # Act
    data = ParserMultiplyRule()
    with pytest.raises(ValueError) as exc_info:
        data.parse_dict(contents, logger)

    exception_raised = exc_info.value

    # Assert
    expected_message = (
        "ERROR in position 0 is type <class 'str'>. "
        "ERROR in position 1 is type <class 'str'>. "
        "Multipliers should be a list of int or floats, "
        "received: ['a', 'b', 2]"
    )
    assert exception_raised.args[0] == expected_message

test_parse_wrong_dict_to_rule_data_logic()

Test if an incorrect dictionary is not parsed

Source code in tests/data/parsers/test_parser_multiply_rule.py

def test_parse_wrong_dict_to_rule_data_logic():
    """Test if an incorrect dictionary is not parsed"""
    # Arrange
    contents = {
        "name": "testname",
        "input_variable": "input",
        "output_variable": "output",
    }
    logger = Mock(ILogger)

    # Act
    data = ParserMultiplyRule()

    with pytest.raises(AttributeError) as exc_info:
        data.parse_dict(contents, logger)

    exception_raised = exc_info.value

    # Assert
    expected_message = "Missing element multipliers_table"
    assert exception_raised.args[0] == expected_message

test_parser_multiply_rule_creation_logic()

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

Source code in tests/data/parsers/test_parser_multiply_rule.py

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

    # Act
    data = ParserMultiplyRule()

    # Assert

    assert isinstance(data, IParserRuleBase)
    assert data.rule_type_name == "multiply_rule"