Narrowing for @final types in match statements

[baluyotraf]

Currently @final types narrow down when using is in an if/else chain:

Playground

from typing import final, assert_never, reveal_type

class Types:

    @final
    class A:
        pass

    @final
    class B:
        pass

def pprint_is(cls: type[Types.A | Types.B]) -> None:
    if cls is Types.A:
        print("A")
        reveal_type(cls)
    elif cls is Types.B:
        print("B")
        reveal_type(cls)
    else:
        reveal_type(cls)
        assert_never(cls)


pprint_is(Types.A)
pprint_is(Types.B)

But it does not seem to happen for match statements, even if they have the same runtime behavior.

Playground

from typing import final, assert_never, reveal_type

class Types:

    @final
    class A:
        pass

    @final
    class B:
        pass

def pprint_match(cls: type[Types.A | Types.B]) -> None:
    match cls:
        case Types.A:
            print("A")
            reveal_type(cls)
        case Types.B:
            print("B")
            reveal_type(cls)
        case _:
            reveal_type(cls)
            assert_never(cls)


pprint_match(Types.A)
pprint_match(Types.B)