Use type checking to check that string is of certain values only

user_3848207 0
Viewed 367

Suppose I have a python function def func_AB(param1: str). param1 can only take on values A or B. If it takes on any other string value, an error should appear.

Is it possible to use python type checking to give out error when this happens? Currently, I use assert to check that param1 contains the valid string value.

I am using python 3.8.5

python-typing python-3.x python
1 Answers

you're looking for Literal

from typing import Literal

def func_AB(param1: Literal['A', 'B']):
    ...
user_2177230
Related