fastest.type.type_inference.infer()

Complexity

Conditions

3

Size

Total Lines	23
Code Lines	14

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

from fastest.type import type_usage_patterns
def infer(variable, statements):
    """
    example: infer("list_var", "def fn():\n\tlist_var = [1]") -> ['list'] #
    example: infer("some_var", "def fn():\n\tsome_var + some_other") -> ['int', 'str'] #
    :param variable:
    :param statements:
    :return:
    """
    statements = statements.split('\n')
    statements = statements[1:]
    type_chances = [0] * 5
    for statement in statements:
        type_chances[INT]   += type_usage_patterns.used_as_int(statement, variable)

The variable INT does not seem to be defined.

        type_chances[STR]   += type_usage_patterns.used_as_str(statement, variable)

The variable STR does not seem to be defined.

        type_chances[LIST]  += type_usage_patterns.used_as_list(statement, variable)

The variable LIST does not seem to be defined.

        type_chances[TUPLE] += type_usage_patterns.used_as_tuple(statement, variable)

The variable TUPLE does not seem to be defined.

        type_chances[DICT]  += type_usage_patterns.used_as_dict(statement, variable)

The variable DICT does not seem to be defined.

    max_prob_type = max(type_chances)
    if type_chances.count(max_prob_type) > 1:
        return [type_map[i] for i, type_chance in enumerate(type_chances) if type_chance == max_prob_type]
    else:
        return [type_map[type_chances.index(max_prob_type)]]