ceph/src/arrow/python/pyarrow/tests/test_strategies.py

   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)