Exhaustiveness Research about sealed-typing-pep HOT 3 CLOSED

from enum import Enum, auto

from typing_extensions import NoReturn


# TODO: Remove the need for this when mypy > 0.942 is released.
#     At that point, simply import this from `typing_extensions`
def assert_never(__arg: NoReturn) -> NoReturn:
    raise AssertionError("Expected code to be unreachable")


class Color(Enum):
    Red = auto()
    Green = auto()
    Blue = auto()


def c(color: Color) -> int:
    match color:
        case Color.Red:
            pass
        case Color.Blue:
            pass
        case _ as u:
            assert_never(u)


def int_or_str(arg: int | str) -> None:
    match arg:
        case int():
            print("It's an int")
        case str():
            print("It's a str")
        case _ as unreachable:
            assert_never(unreachable)

from dataclasses import dataclass
from enum import Enum, auto

# TODO: Remove the need for this when mypy > 0.942 is released.
#     At that point, simply import this from `typing_extensions`
def assert_never(__arg: NoReturn) -> NoReturn:
    raise AssertionError("Expected code to be unreachable")


class UsState(Enum):
    Alabama = auto()
    Alaska = auto()
    # --snip--


@dataclass
class Nickle:
    ...


@dataclass
class Dime:
    ...


@dataclass
class Quarter:
    us_state: UsState


Coin = Nickle | Dime | Quarter


def value_in_cents(coin: Coin) -> int:
    match coin:
        case Nickle():
            return 5
        # try removing one of the `case`s
        case Dime():
            return 10
        case Quarter(state):
            print(f"State quarter from {state}")
            return 25
        case _ as u:
            assert_never(u)