File tree Expand file tree Collapse file tree 2 files changed +14
-5
lines changed Expand file tree Collapse file tree 2 files changed +14
-5
lines changed Original file line number Diff line number Diff line change 6
6
7
7
8
8
class Vector3D (Vector3DSchema ):
9
- pass
9
+ def __init__ (self , root : List [float ]):
10
+ super ().__init__ (root = root )
10
11
11
12
@property
12
13
def value (self ):
13
14
return self .root
14
15
15
16
16
17
class RoundedVector3D (RoundNumericValuesMixin , Vector3D ):
18
+ def __init__ (self , root : List [float ]):
19
+ super ().__init__ (root = root )
20
+
17
21
@model_serializer
18
22
def to_dict (self , skip_rounding : bool = False ) -> List [float ]:
19
23
rounded_value = self .round_array_or_number (self .root ) if not skip_rounding else self .root
Original file line number Diff line number Diff line change 6
6
7
7
8
8
def test_vector_init ():
9
- vector = Vector3D (root = VECTOR_FLOAT )
9
+ vector = Vector3D (VECTOR_FLOAT )
10
10
assert vector .model_dump () == VECTOR_FLOAT
11
11
12
12
@@ -19,23 +19,28 @@ def test_vector_init_wrong_type():
19
19
20
20
def test_vector_init_wrong_size ():
21
21
try :
22
- _ = Vector3D (root = [1 , 2 ])
22
+ _ = Vector3D ([1 , 2 ])
23
23
assert False
24
24
except Exception :
25
25
assert True
26
26
27
27
28
+ def test_rounded_vector_init ():
29
+ vector = RoundedVector3D (VECTOR_FLOAT )
30
+ assert vector .model_dump () == VECTOR_FLOAT
31
+
32
+
28
33
def test_rounded_vector_serialization ():
29
34
class_reference = RoundedVector3D
30
35
class_reference .__round_precision__ = 4
31
- vector = class_reference (root = VECTOR_FLOAT )
36
+ vector = class_reference (VECTOR_FLOAT )
32
37
assert vector .model_dump () == VECTOR_FLOAT_ROUNDED_4
33
38
assert vector .value_rounded == VECTOR_FLOAT_ROUNDED_4
34
39
assert vector .value == VECTOR_FLOAT
35
40
36
41
class_reference = RoundedVector3D
37
42
class_reference .__round_precision__ = 3
38
- vector = class_reference (root = VECTOR_FLOAT )
43
+ vector = class_reference (VECTOR_FLOAT )
39
44
assert vector .model_dump () == VECTOR_FLOAT_ROUNDED_3
40
45
assert vector .value_rounded == VECTOR_FLOAT_ROUNDED_3
41
46
assert vector .value == VECTOR_FLOAT
You can’t perform that action at this time.
0 commit comments