Skip to content

Commit bdff0eb

Browse files
committed
add tests for validatedDataclass1
Signed-off-by: Zen <[email protected]>
1 parent 1c58f78 commit bdff0eb

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

tests/test_validated_dataclass.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
from unittest import TestCase, main
2+
3+
from zenlib.types import validatedDataclass
4+
5+
6+
@validatedDataclass
7+
class testDataClass:
8+
a: int = None
9+
b: str = None
10+
11+
12+
class TestValidatedDataclass(TestCase):
13+
def test_validated_dataclass(self):
14+
c = testDataClass()
15+
c.a = 1
16+
c.b = "test"
17+
self.assertTrue(hasattr(c, "logger"))
18+
19+
def test_bad_type(self):
20+
c = testDataClass()
21+
with self.assertRaises(TypeError):
22+
c.a = "test"
23+
24+
25+
if __name__ == "__main__":
26+
main()

0 commit comments

Comments
 (0)