FM / arrow

/usr/local/lib64/python3.6/site-packages/pandas/tests/extension/arrow/

/usr/local/lib64/python3.6/site-packages/pandas/tests/extension/arrow

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
arrays.py	5360	0644	edit dl rm
test_bool.py	2904	0644	edit dl rm
test_string.py	301	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/extension/arrow/test_bool.py (2904B)

import numpy as np
import pytest

from pandas.compat import PY37

import pandas as pd
import pandas._testing as tm
from pandas.tests.extension import base

pytest.importorskip("pyarrow", minversion="0.13.0")

from .arrays import ArrowBoolArray, ArrowBoolDtype  # isort:skip

@pytest.fixture
def dtype():
    return ArrowBoolDtype()

@pytest.fixture
def data():
    values = np.random.randint(0, 2, size=100, dtype=bool)
    values[1] = ~values[0]
    return ArrowBoolArray.from_scalars(values)

@pytest.fixture
def data_missing():
    return ArrowBoolArray.from_scalars([None, True])

def test_basic_equals(data):
    # https://github.com/pandas-dev/pandas/issues/34660
    assert pd.Series(data).equals(pd.Series(data))

class BaseArrowTests:
    pass

class TestDtype(BaseArrowTests, base.BaseDtypeTests):
    def test_array_type_with_arg(self, data, dtype):
        pytest.skip("GH-22666")

class TestInterface(BaseArrowTests, base.BaseInterfaceTests):
    def test_copy(self, data):
        # __setitem__ does not work, so we only have a smoke-test
        data.copy()

def test_view(self, data):
        # __setitem__ does not work, so we only have a smoke-test
        data.view()

class TestConstructors(BaseArrowTests, base.BaseConstructorsTests):
    def test_from_dtype(self, data):
        pytest.skip("GH-22666")

# seems like some bug in isna on empty BoolArray returning floats.
    @pytest.mark.xfail(reason="bad is-na for empty data")
    def test_from_sequence_from_cls(self, data):
        super().test_from_sequence_from_cls(data)

@pytest.mark.skipif(not PY37, reason="timeout on Linux py36_locale")
    @pytest.mark.xfail(reason="pa.NULL is not recognised as scalar, GH-33899")
    def test_series_constructor_no_data_with_index(self, dtype, na_value):
        # pyarrow.lib.ArrowInvalid: only handle 1-dimensional arrays
        super().test_series_constructor_no_data_with_index(dtype, na_value)

@pytest.mark.skipif(not PY37, reason="timeout on Linux py36_locale")
    @pytest.mark.xfail(reason="pa.NULL is not recognised as scalar, GH-33899")
    def test_series_constructor_scalar_na_with_index(self, dtype, na_value):
        # pyarrow.lib.ArrowInvalid: only handle 1-dimensional arrays
        super().test_series_constructor_scalar_na_with_index(dtype, na_value)

@pytest.mark.xfail(reason="raises AssertionError")
    def test_construct_empty_dataframe(self, dtype):
        super().test_construct_empty_dataframe(dtype)

class TestReduce(base.BaseNoReduceTests):
    def test_reduce_series_boolean(self):
        pass

class TestReduceBoolean(base.BaseBooleanReduceTests):
    pass

def test_is_bool_dtype(data):
    assert pd.api.types.is_bool_dtype(data)
    assert pd.core.common.is_bool_indexer(data)
    s = pd.Series(range(len(data)))
    result = s[data]
    expected = s[np.asarray(data)]
    tm.assert_series_equal(result, expected)

import numpy as np import pytest from pandas.compat import PY37 import pandas as pd import pandas._testing as tm from pandas.tests.extension import base pytest.importorskip("pyarrow", minversion="0.13.0") from .arrays import ArrowBoolArray, ArrowBoolDtype # isort:skip @pytest.fixture def dtype(): return ArrowBoolDtype() @pytest.fixture def data(): values = np.random.randint(0, 2, size=100, dtype=bool) values[1] = ~values[0] return ArrowBoolArray.from_scalars(values) @pytest.fixture def data_missing(): return ArrowBoolArray.from_scalars([None, True]) def test_basic_equals(data): # https://github.com/pandas-dev/pandas/issues/34660 assert pd.Series(data).equals(pd.Series(data)) class BaseArrowTests: pass class TestDtype(BaseArrowTests, base.BaseDtypeTests): def test_array_type_with_arg(self, data, dtype): pytest.skip("GH-22666") class TestInterface(BaseArrowTests, base.BaseInterfaceTests): def test_copy(self, data): # __setitem__ does not work, so we only have a smoke-test data.copy() def test_view(self, data): # __setitem__ does not work, so we only have a smoke-test data.view() class TestConstructors(BaseArrowTests, base.BaseConstructorsTests): def test_from_dtype(self, data): pytest.skip("GH-22666") # seems like some bug in isna on empty BoolArray returning floats. @pytest.mark.xfail(reason="bad is-na for empty data") def test_from_sequence_from_cls(self, data): super().test_from_sequence_from_cls(data) @pytest.mark.skipif(not PY37, reason="timeout on Linux py36_locale") @pytest.mark.xfail(reason="pa.NULL is not recognised as scalar, GH-33899") def test_series_constructor_no_data_with_index(self, dtype, na_value): # pyarrow.lib.ArrowInvalid: only handle 1-dimensional arrays super().test_series_constructor_no_data_with_index(dtype, na_value) @pytest.mark.skipif(not PY37, reason="timeout on Linux py36_locale") @pytest.mark.xfail(reason="pa.NULL is not recognised as scalar, GH-33899") def test_series_constructor_scalar_na_with_index(self, dtype, na_value): # pyarrow.lib.ArrowInvalid: only handle 1-dimensional arrays super().test_series_constructor_scalar_na_with_index(dtype, na_value) @pytest.mark.xfail(reason="raises AssertionError") def test_construct_empty_dataframe(self, dtype): super().test_construct_empty_dataframe(dtype) class TestReduce(base.BaseNoReduceTests): def test_reduce_series_boolean(self): pass class TestReduceBoolean(base.BaseBooleanReduceTests): pass def test_is_bool_dtype(data): assert pd.api.types.is_bool_dtype(data) assert pd.core.common.is_bool_indexer(data) s = pd.Series(range(len(data))) result = s[data] expected = s[np.asarray(data)] tm.assert_series_equal(result, expected)