nested-polymorphism-overload-py

switch.py

class Ball:
    pass


class Game:
    pass


class Tantrum:
    pass


class Athlete:
    def throw(self, thing):
        kind = type(thing)
        if kind == Ball:
            return "⚾"
        elif kind == Game:
            return "🤞"
        elif kind == Tantrum:
            return "👿"
        else:
            raise Exception(f"don't know how to throw thing of type {kind}")

test_switch.py

from switch import *


def test_athlete_throw():
    athlete = Athlete()
    assert "⚾" == athlete.throw(Ball())
    assert "🤞" == athlete.throw(Game())
    assert "👿" == athlete.throw(Tantrum())

    try:
        athlete.throw(42)
        assert False, "expected exception"
    except:
        pass