Alle Dateien aus dem Pythonkurs
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

51 lines
1.8 KiB

import pytest
import pandas.core.dtypes.concat as _concat
import pandas as pd
from pandas import Series
import pandas._testing as tm
def test_concat_mismatched_categoricals_with_empty():
# concat_compat behavior on series._values should match pd.concat on series
ser1 = Series(["a", "b", "c"], dtype="category")
ser2 = Series([], dtype="category")
msg = "The behavior of array concatenation with empty entries is deprecated"
with tm.assert_produces_warning(FutureWarning, match=msg):
result = _concat.concat_compat([ser1._values, ser2._values])
with tm.assert_produces_warning(FutureWarning, match=msg):
expected = pd.concat([ser1, ser2])._values
tm.assert_categorical_equal(result, expected)
@pytest.mark.parametrize("copy", [True, False])
def test_concat_single_dataframe_tz_aware(copy):
# https://github.com/pandas-dev/pandas/issues/25257
df = pd.DataFrame(
{"timestamp": [pd.Timestamp("2020-04-08 09:00:00.709949+0000", tz="UTC")]}
)
expected = df.copy()
result = pd.concat([df], copy=copy)
tm.assert_frame_equal(result, expected)
def test_concat_periodarray_2d():
pi = pd.period_range("2016-01-01", periods=36, freq="D")
arr = pi._data.reshape(6, 6)
result = _concat.concat_compat([arr[:2], arr[2:]], axis=0)
tm.assert_period_array_equal(result, arr)
result = _concat.concat_compat([arr[:, :2], arr[:, 2:]], axis=1)
tm.assert_period_array_equal(result, arr)
msg = (
"all the input array dimensions.* for the concatenation axis must match exactly"
)
with pytest.raises(ValueError, match=msg):
_concat.concat_compat([arr[:, :2], arr[:, 2:]], axis=0)
with pytest.raises(ValueError, match=msg):
_concat.concat_compat([arr[:2], arr[2:]], axis=1)