]>
Commit | Line | Data |
---|---|---|
1d09f67e TL |
1 | # Licensed to the Apache Software Foundation (ASF) under one |
2 | # or more contributor license agreements. See the NOTICE file | |
3 | # distributed with this work for additional information | |
4 | # regarding copyright ownership. The ASF licenses this file | |
5 | # to you under the Apache License, Version 2.0 (the | |
6 | # "License"); you may not use this file except in compliance | |
7 | # with the License. You may obtain a copy of the License at | |
8 | # | |
9 | # http://www.apache.org/licenses/LICENSE-2.0 | |
10 | # | |
11 | # Unless required by applicable law or agreed to in writing, | |
12 | # software distributed under the License is distributed on an | |
13 | # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
14 | # KIND, either express or implied. See the License for the | |
15 | # specific language governing permissions and limitations | |
16 | # under the License. | |
17 | ||
18 | import hypothesis as h | |
19 | ||
20 | import pyarrow as pa | |
21 | import pyarrow.tests.strategies as past | |
22 | ||
23 | ||
24 | @h.given(past.all_types) | |
25 | def test_types(ty): | |
26 | assert isinstance(ty, pa.lib.DataType) | |
27 | ||
28 | ||
29 | @h.given(past.all_fields) | |
30 | def test_fields(field): | |
31 | assert isinstance(field, pa.lib.Field) | |
32 | ||
33 | ||
34 | @h.given(past.all_schemas) | |
35 | def test_schemas(schema): | |
36 | assert isinstance(schema, pa.lib.Schema) | |
37 | ||
38 | ||
39 | @h.given(past.all_arrays) | |
40 | def test_arrays(array): | |
41 | assert isinstance(array, pa.lib.Array) | |
42 | ||
43 | ||
44 | @h.given(past.arrays(past.primitive_types, nullable=False)) | |
45 | def test_array_nullability(array): | |
46 | assert array.null_count == 0 | |
47 | ||
48 | ||
49 | @h.given(past.all_chunked_arrays) | |
50 | def test_chunked_arrays(chunked_array): | |
51 | assert isinstance(chunked_array, pa.lib.ChunkedArray) | |
52 | ||
53 | ||
54 | @h.given(past.all_record_batches) | |
55 | def test_record_batches(record_bath): | |
56 | assert isinstance(record_bath, pa.lib.RecordBatch) | |
57 | ||
58 | ||
59 | @h.given(past.all_tables) | |
60 | def test_tables(table): | |
61 | assert isinstance(table, pa.lib.Table) |