CoCalc -- test_structs.py

GitHub Repository: pola-rs/polars
Path: blob/main/py-polars/tests/unit/constructors/test_structs.py
⁶⁹³⁹ views
1
import polars as pl
2

3

4
def test_constructor_non_strict_schema_17956() -> None:
5
    schema = {
6
        "logged_event": pl.Struct(
7
            [
8
                pl.Field(
9
                    "completetask",
10
                    pl.Struct(
11
                        [
12
                            pl.Field(
13
                                "parameters",
14
                                pl.List(
15
                                    pl.Struct(
16
                                        [
17
                                            pl.Field(
18
                                                "numericarray",
19
                                                pl.Struct(
20
                                                    [
21
                                                        pl.Field(
22
                                                            "value", pl.List(pl.Float64)
23
                                                        ),
24
                                                    ]
25
                                                ),
26
                                            ),
27
                                        ]
28
                                    )
29
                                ),
30
                            ),
31
                        ]
32
                    ),
33
                ),
34
            ]
35
        ),
36
    }
37

38
    data = {
39
        "logged_event": {
40
            "completetask": {
41
                "parameters": [
42
                    {
43
                        "numericarray": {
44
                            "value": [
45
                                431,
46
                                431,
47
                                431,
48
                                431,
49
                                431,
50
                                431,
51
                                431,
52
                                431,
53
                                431,
54
                                431,
55
                                431,
56
                                431,
57
                                431,
58
                                431,
59
                                431,
60
                                431,
61
                                431,
62
                                431,
63
                                431,
64
                                431,
65
                                431,
66
                                431,
67
                                431,
68
                                431,
69
                                431,
70
                                431,
71
                                431,
72
                                431,
73
                                431,
74
                                431,
75
                                431,
76
                                431,
77
                                431,
78
                                431,
79
                                430.5,
80
                                431,
81
                                431,
82
                                431,
83
                            ]
84
                        }
85
                    }
86
                ]
87
            }
88
        }
89
    }
90

91
    lazyframe = pl.LazyFrame(
92
        [data],
93
        schema=schema,
94
        strict=False,
95
    )
96
    assert lazyframe.collect().to_dict(as_series=False) == {
97
        "logged_event": [
98
            {
99
                "completetask": {
100
                    "parameters": [
101
                        {
102
                            "numericarray": {
103
                                "value": [
104
                                    431.0,
105
                                    431.0,
106
                                    431.0,
107
                                    431.0,
108
                                    431.0,
109
                                    431.0,
110
                                    431.0,
111
                                    431.0,
112
                                    431.0,
113
                                    431.0,
114
                                    431.0,
115
                                    431.0,
116
                                    431.0,
117
                                    431.0,
118
                                    431.0,
119
                                    431.0,
120
                                    431.0,
121
                                    431.0,
122
                                    431.0,
123
                                    431.0,
124
                                    431.0,
125
                                    431.0,
126
                                    431.0,
127
                                    431.0,
128
                                    431.0,
129
                                    431.0,
130
                                    431.0,
131
                                    431.0,
132
                                    431.0,
133
                                    431.0,
134
                                    431.0,
135
                                    431.0,
136
                                    431.0,
137
                                    431.0,
138
                                    430.5,
139
                                    431.0,
140
                                    431.0,
141
                                    431.0,
142
                                ]
143
                            }
144
                        }
145
                    ]
146
                }
147
            }
148
        ]
149
    }
150

151
Product

Resources

Company
