Skip to content

Commit 575ea5e

Browse files
committed
Fix up tests for py3.10. always set pydantic description on csp.Enum
Signed-off-by: Rob Ambalu <robert.ambalu@point72.com>
1 parent 056c5f9 commit 575ea5e

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

csp/impl/enum.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def __get_pydantic_json_schema__(cls, _core_schema, handler):
4949
field_schema.update(
5050
type="string",
5151
title=cls.__name__,
52-
description=cls.__doc__ or "An enumeration of {}".format(cls.__name__),
52+
description="An enumeration of {}".format(cls.__name__),
5353
enum=list(cls.__members__.keys()),
5454
)
5555
return field_schema

csp/tests/impl/test_enum.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,13 +164,11 @@ def test_subclassing(self):
164164
class A(csp.Enum):
165165
RED = 1
166166

167-
with self.assertRaises(TypeError) as cm:
167+
with self.assertRaisesRegex(TypeError, "cannot extend"):
168168

169169
class B(A):
170170
GREEN = 2
171171

172-
self.assertEqual("<enum 'B'> cannot extend <enum 'A'>", str(cm.exception))
173-
174172
def test_pydantic_validation(self):
175173
assert MyModel(enum="FIELD2").enum == MyEnum3.FIELD2
176174
assert MyModel(enum=0).enum == MyEnum3.FIELD1

0 commit comments

Comments
 (0)