Nested if Optional checking for class attributes fails

Checking optional attributes in a class does not persist in a nested if:

(test.py)

#!/usr/bin/env python3

import typing


class Example:
    def __init__(self):
        self.a: typing.Optional[int]


def check_class(foo: Example, bar: Example):
    reveal_type(foo.a)
    if foo.a is not None:
        reveal_type(foo.a)
        if foo.a > 0:
            reveal_type(foo.a)
            foo = bar
        elif foo.a < 0:
            reveal_type(foo.a)
            foo.a += 1

Output:

$ mypy test.py
test.py:12: note: Revealed type is "Union[builtins.int, None]"
test.py:14: note: Revealed type is "builtins.int"
test.py:16: note: Revealed type is "builtins.int"
test.py:18: error: Unsupported operand types for > ("int" and "None")
test.py:18: note: Left operand is of type "Optional[int]"
test.py:19: note: Revealed type is "Union[builtins.int, None]"
test.py:20: error: Unsupported operand types for + ("None" and "int")
test.py:20: note: Left operand is of type "Optional[int]"

Environment

• Python 3.9.9
• mypy == 0.971
• Linux x86_64
• No other mypy config

Beginne damit, das bereitgestellte test.py-Beispiel mit mypy auszuführen, und prüfe die reveal_type-Ausgabe sowie die gemeldeten Fehler in den verschachtelten Verzweigungen. Die Arbeit ist abgeschlossen, wenn die beabsichtigte Narrowing-Behandlung von non-None dort konsistent funktioniert, ohne nach einer erneuten Zuweisung an foo ein falsches Narrowing einzuführen.