python/pyarrow/tests/test_cpp_internals.py

# 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)