Skip to content

Commit f6eaecf

Browse files
committed
feat: updates enum unit tests
1 parent 809afe4 commit f6eaecf

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_enum_models.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,10 +104,24 @@ class B(AindModel):
104104

105105
a: EnumSubset[TestEnum.FOO, Manufacturer.OTHER]
106106

107+
with self.assertRaises(Exception) as e3:
108+
109+
class C(AindModel):
110+
"""temp model"""
111+
112+
a: EnumSubset[TestEnum.FOO]
113+
114+
C(a=TestEnum.BAR)
115+
107116
expected_exception1 = """TypeError("Only Enums are allowed. <class 'int'>")"""
108117
expected_exception2 = """ValueError('All enums must be of the same class.')"""
118+
expected_exception3 = (
119+
"ValidationError(model='C', errors=[{'loc': ('a',), 'msg': 'Value not allowed! TestEnum.BAR', "
120+
"'type': 'value_error'}])"
121+
)
109122
self.assertEqual(expected_exception1, repr(e1.exception))
110123
self.assertEqual(expected_exception2, repr(e2.exception))
124+
self.assertEqual(expected_exception3, repr(e3.exception))
111125

112126

113127
if __name__ == "__main__":

0 commit comments

Comments
 (0)