58 lines
2 KiB
Python
58 lines
2 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import os.path
|
|
from os.path import join as pjoin
|
|
|
|
import pytest
|
|
|
|
from pyarrow._pyarrow_cpp_tests import get_cpp_tests
|
|
|
|
|
|
def inject_cpp_tests(ns):
|
|
"""
|
|
Inject C++ tests as Python functions into namespace `ns` (a dict).
|
|
"""
|
|
for case in get_cpp_tests():
|
|
|
|
def wrapper(case=case):
|
|
case()
|
|
wrapper.__name__ = wrapper.__qualname__ = case.name
|
|
wrapper.__module__ = ns['__name__']
|
|
# Add numpy or pandas marks if the test requires it
|
|
if 'numpy' in case.name:
|
|
wrapper = pytest.mark.numpy(wrapper)
|
|
elif 'pandas' in case.name:
|
|
wrapper = pytest.mark.pandas(wrapper)
|
|
ns[case.name] = wrapper
|
|
|
|
|
|
inject_cpp_tests(globals())
|
|
|
|
|
|
def test_pyarrow_include():
|
|
# We need to make sure that pyarrow/include is always
|
|
# created. Either with PyArrow C++ header files or with
|
|
# Arrow C++ and PyArrow C++ header files together
|
|
|
|
source = os.path.dirname(os.path.abspath(__file__))
|
|
pyarrow_dir = pjoin(source, '..')
|
|
pyarrow_include = pjoin(pyarrow_dir, 'include')
|
|
pyarrow_cpp_include = pjoin(pyarrow_include, 'arrow', 'python')
|
|
|
|
assert os.path.exists(pyarrow_include)
|
|
assert os.path.exists(pyarrow_cpp_include)
|