Python 3.10+ Structural Pattern Matching

structural_pattern_matching.py

from __future__ import annotations
from typing import NamedTuple


class Point2D(NamedTuple):
    x: int
    y: int


class Point3D(NamedTuple):
    x: int
    y: int
    z: int


def make_point_3d_old(point: tuple[int, ...] | Point2D) -> Point3D:
    if isinstance(point, Point2D):
        return Point3D(point.x, point.y, 0)
    elif isinstance(point, Point3D):
        return point
    elif isinstance(point, tuple):
        if len(point) == 2:
            return Point3D(point[0], point[1], 0)
        elif len(point) == 3:
            return Point3D(point[0], point[1], point[2])
        else:
            raise TypeError("Requires tuple length l: 1 < l <= 3")
    else:
        raise TypeError("Not a supported point type")
    

def make_point_3d(point: tuple[int, ...] | Point2D) -> Point3D:
    match point:
        case (x, y):
            return Point3D(x, y, 0)
        case (x, y, z):
            return Point3D(x, y, z)
        case Point2D(x, y):
            return Point3D(x, y, 0)
        case Point3D(_, _, _):
            return point
        case _:
            raise TypeError("Not a supported point type")