FM / extension

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

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

Name	Size	Mode	Actions
arrow/	-	0755	rm
base/	-	0755	rm
decimal/	-	0755	rm
json/	-	0755	rm
list/	-	0755	rm
__pycache__/	-	0755	rm
conftest.py	3785	0644	edit dl rm
test_boolean.py	12869	0644	edit dl rm
test_categorical.py	8476	0644	edit dl rm
test_common.py	2091	0644	edit dl rm
test_datetime.py	6610	0644	edit dl rm
test_external_block.py	1188	0644	edit dl rm
test_integer.py	7327	0644	edit dl rm
test_interval.py	4202	0644	edit dl rm
test_numpy.py	16996	0644	edit dl rm
test_period.py	4480	0644	edit dl rm
test_sparse.py	14403	0644	edit dl rm
test_string.py	2671	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/extension/test_external_block.py (1188B)

import numpy as np
import pytest

import pandas as pd
from pandas.core.internals import BlockManager
from pandas.core.internals.blocks import ExtensionBlock

class CustomBlock(ExtensionBlock):

_holder = np.ndarray
    _can_hold_na = False

def concat_same_type(self, to_concat, placement=None):
        """
        Always concatenate disregarding self.ndim as the values are
        always 1D in this custom Block
        """
        values = np.concatenate([blk.values for blk in to_concat])
        placement = self.mgr_locs if self.ndim == 2 else slice(len(values))
        return self.make_block_same_class(values, placement=placement)

@pytest.fixture
def df():
    df1 = pd.DataFrame({"a": [1, 2, 3]})
    blocks = df1._mgr.blocks
    values = np.arange(3, dtype="int64")
    custom_block = CustomBlock(values, placement=slice(1, 2))
    blocks = blocks + (custom_block,)
    block_manager = BlockManager(blocks, [pd.Index(["a", "b"]), df1.index])
    return pd.DataFrame(block_manager)

def test_concat_axis1(df):
    # GH17954
    df2 = pd.DataFrame({"c": [0.1, 0.2, 0.3]})
    res = pd.concat([df, df2], axis=1)
    assert isinstance(res._mgr.blocks[1], CustomBlock)

import numpy as np import pytest import pandas as pd from pandas.core.internals import BlockManager from pandas.core.internals.blocks import ExtensionBlock class CustomBlock(ExtensionBlock): _holder = np.ndarray _can_hold_na = False def concat_same_type(self, to_concat, placement=None): """ Always concatenate disregarding self.ndim as the values are always 1D in this custom Block """ values = np.concatenate([blk.values for blk in to_concat]) placement = self.mgr_locs if self.ndim == 2 else slice(len(values)) return self.make_block_same_class(values, placement=placement) @pytest.fixture def df(): df1 = pd.DataFrame({"a": [1, 2, 3]}) blocks = df1._mgr.blocks values = np.arange(3, dtype="int64") custom_block = CustomBlock(values, placement=slice(1, 2)) blocks = blocks + (custom_block,) block_manager = BlockManager(blocks, [pd.Index(["a", "b"]), df1.index]) return pd.DataFrame(block_manager) def test_concat_axis1(df): # GH17954 df2 = pd.DataFrame({"c": [0.1, 0.2, 0.3]}) res = pd.concat([df, df2], axis=1) assert isinstance(res._mgr.blocks[1], CustomBlock)