From 87d850ca67df20db3e10b7e1c05d415c03f21e66 Mon Sep 17 00:00:00 2001 From: Adam Dangoor Date: Mon, 4 Nov 2024 16:16:45 +0000 Subject: [PATCH] Use @unique to ensure and communicate enum uniqueness --- src/mock_vws/_constants.py | 4 +++- src/mock_vws/states.py | 3 ++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/mock_vws/_constants.py b/src/mock_vws/_constants.py index 25753ed58..61443bbc7 100644 --- a/src/mock_vws/_constants.py +++ b/src/mock_vws/_constants.py @@ -2,12 +2,13 @@ Constants used to make the VWS mock. """ -from enum import Enum +from enum import Enum, unique from beartype import beartype @beartype +@unique class ResultCodes(Enum): """Constants representing various VWS result codes. @@ -42,6 +43,7 @@ class ResultCodes(Enum): @beartype +@unique class TargetStatuses(Enum): """Constants representing VWS target statuses. diff --git a/src/mock_vws/states.py b/src/mock_vws/states.py index c3f55b019..7233fd8c9 100644 --- a/src/mock_vws/states.py +++ b/src/mock_vws/states.py @@ -2,12 +2,13 @@ Vuforia database states. """ -from enum import StrEnum, auto +from enum import StrEnum, auto, unique from beartype import beartype @beartype +@unique class States(StrEnum): """ Constants representing various web service states.