BLOCKER: LPython can't find dict key, but CPython can find it

[rebcabin]

repro: https://github.com/rebcabin/lpython/tree/brian-lasr/lasr/LP-pycharm/Issue2011

offending code:

I4C: dict[str, i32] = {
    '0': 0, '1': 1, '2': 2, '3': 3,
    '4': 4, '5': 5, '6': 6, '7': 7,
    '8': 8, '9': 9,
    'a': 10, 'b': 11, 'c': 12, 'd': 13,
    'A': 10, 'B': 11, 'C': 12, 'D': 13,
    'e': 14, 'f': 15,
    'E': 14, 'F': 15}


def cnvi(s : str, base : i32=10) -> i32:
    """Assume input has been through 'match_integer'."""
    assert base == 10 or base == 8 or base == 16 or base == 2
    result : i32 = 0
    c : str
    t : str = strip_numerical_noise(s)
    pow: i32 = base ** (len(t) - 1)
    for c in t:
        incr : i32 = pow * I4C[c]
        result += incr
        pow = (pow // base)
    return result


if __name__ == '__main__':
    assert cnvi('0b0', base=2) == 0
    assert cnvi('0b1', base=2) == 1
    assert cnvi('0b10', base=2) == 2
    assert cnvi('0b11', base=2) == 3
    assert cnvi('0b1111_0100_111', base=2) == 1959
    assert cnvi('0b7a7', base=16) == 1959  ############# ATTENTION ##############

runs:

(lp) ┌─(~/CLionProjects/lpython/lasr/LP-pycharm/Issue2011)───────────────────────────────────────────────────────────────────────────────────────────────(brian@Golf37:s012)─┐
└─(17:51:22 on brian-lasr ✹ ✭)──> PYTHONPATH='../../../src/runtime/lpython' python lasr_lexer.py                                                          ──(Thu,Jun22)─┘
(lp) ┌─(~/CLionProjects/lpython/lasr/LP-pycharm/Issue2011)───────────────────────────────────────────────────────────────────────────────────────────────(brian@Golf37:s012)─┐
└─(17:51:30 on brian-lasr ✹ ✭)──> ~/CLionProjects/lpython/src/bin/lpython -I. lasr_lexer.py                                                               ──(Thu,Jun22)─┘
KeyError: The dict does not contain the specified key

[certik]

Here is a minimal reproducible example:

from lpython import i32

def main():
    d: dict[str, i32] = {
        '2': 2, '3': 3,                
        '4': 4, '5': 5, '6': 6, '7': 7,
        '8': 8, '9': 9,
        'a': 10, 'b': 11, 'c': 12, 'd': 13,
        'A': 10, 'B': 11, 'C': 12, 'D': 13,
        'e': 14, 'f': 15,
        'E': 14, 'F': 15}
    print(d['a'])

main()

Gives:

$ PYTHONPATH=src/runtime/lpython python a.py          
10
$ lpython a.py                              
KeyError: The dict does not contain the specified key
$ lpython --backend=c a.py   
10

If I remove any more keys from the dictionary, then it passes. It only happens in the LLVM backend, C works.

[certik]

@czgdp1807 do you think you could please have a look at this? This looks like a bug in our dictionary implementation.

[rebcabin]

thanks for jumping on this :) you guys are great!

[rebcabin]

verified ! :)

[certik]

Keep reporting bugs!