""" Tests for the pseudo-public API implemented in internals/api.py and exposed in core.internals """ import pandas as pd from pandas.core import internals from pandas.core.internals import api def test_internals_api(): assert internals.make_block is api.make_block def test_namespace(): # SUBJECT TO CHANGE modules = [ "blocks", "concat", "managers", "construction", "array_manager", "base", "api", "ops", ] expected = [ "Block", "DatetimeTZBlock", "ExtensionBlock", "make_block", "DataManager", "ArrayManager", "BlockManager", "SingleDataManager", "SingleBlockManager", "SingleArrayManager", "concatenate_managers", "create_block_manager_from_blocks", ] result = [x for x in dir(internals) if not x.startswith("__")] assert set(result) == set(expected + modules) def test_make_block_2d_with_dti(): # GH#41168 dti = pd.date_range("2012", periods=3, tz="UTC") blk = api.make_block(dti, placement=[0]) assert blk.shape == (1, 3) assert blk.values.shape == (1, 3)