@@ -65,17 +65,17 @@ class ADataClassWithNullable:
65
65
class ADataClassWithList :
66
66
a : list [int ]
67
67
68
+
68
69
class Choices (enum .Enum ):
69
- A :str = "A"
70
- B :str = "B"
71
- C :str = "C"
72
- D :str = "D"
70
+ A : str = "A"
71
+ B : str = "B"
72
+ C : str = "C"
73
+ D : str = "D"
73
74
74
75
75
76
@dataclasses .dataclass
76
77
class HasEnum :
77
- choice :Choices
78
-
78
+ choice : Choices
79
79
80
80
81
81
class UnitTests (parameterized .TestCase ):
@@ -549,20 +549,25 @@ def b():
549
549
},
550
550
),
551
551
],
552
- ['enum' , Choices , protos .Schema (type = protos .Type .STRING , enum = ['A' , 'B' , 'C' , 'D' ])],
553
- ['enum_list' ,
554
- list [Choices ],
555
- protos .Schema (
556
- type = "ARRAY" ,
557
- items = protos .Schema (type = protos .Type .STRING , enum = ['A' , 'B' , 'C' , 'D' ]),
558
- ),
552
+ ["enum" , Choices , protos .Schema (type = protos .Type .STRING , enum = ["A" , "B" , "C" , "D" ])],
553
+ [
554
+ "enum_list" ,
555
+ list [Choices ],
556
+ protos .Schema (
557
+ type = "ARRAY" ,
558
+ items = protos .Schema (type = protos .Type .STRING , enum = ["A" , "B" , "C" , "D" ]),
559
+ ),
560
+ ],
561
+ [
562
+ "has_enum" ,
563
+ HasEnum ,
564
+ protos .Schema (
565
+ type = protos .Type .OBJECT ,
566
+ properties = {
567
+ "choice" : protos .Schema (type = protos .Type .STRING , enum = ["A" , "B" , "C" , "D" ])
568
+ },
569
+ ),
559
570
],
560
- ['has_enum' ,
561
- HasEnum ,
562
- protos .Schema (
563
- type = protos .Type .OBJECT ,
564
- properties = {'choice' : protos .Schema (type = protos .Type .STRING , enum = ['A' , 'B' , 'C' , 'D' ])})
565
- ]
566
571
)
567
572
def test_auto_schema (self , annotation , expected ):
568
573
def fun (a : annotation ):
0 commit comments