Skip to content

Commit e8e89b2

Browse files
committed
readd nested class test for coverage
1 parent 4d7aba8 commit e8e89b2

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

tests/legacy/test_nestedclass.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import sys
33
from enum import auto
44
from unittest import TestCase
5+
import warnings
56

67
from enum_properties import (
78
EnumProperties,
@@ -279,3 +280,31 @@ class Type3:
279280
self.assertTrue(MyEnum.Type1.value().__class__ is MyEnum.Type1.value)
280281
self.assertTrue(MyEnum.Type2.value().__class__ is MyEnum.Type2.value)
281282
self.assertTrue(MyEnum.Type3.value().__class__ is MyEnum.Type3.value)
283+
284+
def test_unmarked_nested_classes(self):
285+
with warnings.catch_warnings():
286+
warnings.simplefilter("ignore", DeprecationWarning)
287+
288+
class TestEnum(EnumProperties, s("label")):
289+
VALUE1 = auto(), "value1"
290+
VALUE2 = auto(), "value2"
291+
VALUE3 = auto(), "value3"
292+
293+
class NestedClass:
294+
@property
295+
def prop(self):
296+
return "nested"
297+
298+
self.assertEqual(TestEnum.NestedClass().prop, "nested")
299+
self.assertEqual(
300+
[en for en in TestEnum],
301+
[TestEnum.VALUE1, TestEnum.VALUE2, TestEnum.VALUE3],
302+
)
303+
self.assertEqual(
304+
[en.label for en in TestEnum],
305+
[TestEnum.VALUE1.label, TestEnum.VALUE2.label, TestEnum.VALUE3.label],
306+
)
307+
self.assertEqual(
308+
[en for en in TestEnum],
309+
[TestEnum("value1"), TestEnum("value2"), TestEnum("value3")],
310+
)

0 commit comments

Comments
 (0)