Beta Shell v2.0

File Manager

~ /usr/local/python-3.12/lib/python3.12/site-packages/pandas/tests/series/methods

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	225 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:40:26	Del
test_add_prefix_suffix.py	1.52 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_align.py	7.52 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_argsort.py	2.8 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_asof.py	6.18 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_astype.py	24.88 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_autocorr.py	1015 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_between.py	2.52 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_case_when.py	4.12 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_clip.py	5.1 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_combine.py	627 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_combine_first.py	5.29 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_compare.py	4.55 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_convert_dtypes.py	9.5 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_copy.py	3.09 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_count.py	938 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_cov_corr.py	5.58 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_describe.py	6.49 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_diff.py	2.48 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_drop.py	3.31 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_drop_duplicates.py	9.02 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_dropna.py	3.49 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_dtypes.py	209 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_duplicated.py	2.01 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_equals.py	4.08 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_explode.py	4.99 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_fillna.py	39.05 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_get_numeric_data.py	1.15 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_head_tail.py	343 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_infer_objects.py	1.86 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_info.py	4.79 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_interpolate.py	33.46 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_is_monotonic.py	838 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_is_unique.py	953 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_isin.py	8.15 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_isna.py	940 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_item.py	1.59 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_map.py	18.33 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_matmul.py	2.7 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_nlargest.py	8.24 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_nunique.py	481 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_pct_change.py	4.42 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_pop.py	295 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_quantile.py	8.09 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_rank.py	17.68 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_reindex.py	14.23 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_reindex_like.py	1.22 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_rename.py	6.17 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_rename_axis.py	1.48 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_repeat.py	1.24 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_replace.py	31.19 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_reset_index.py	7.66 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_round.py	2.59 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_searchsorted.py	2.43 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_set_name.py	595 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_size.py	566 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_sort_index.py	12.34 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_sort_values.py	9.15 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_to_csv.py	6.34 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_to_dict.py	1.15 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_to_frame.py	1.95 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_to_numpy.py	1.29 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_tolist.py	1.09 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_truncate.py	2.25 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_tz_localize.py	4.23 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_unique.py	2.17 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_unstack.py	5 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_update.py	5.21 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_value_counts.py	9.87 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_values.py	747 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_view.py	1.81 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del

Editing: test_info.py (4.79 KB)

from io import StringIO
from string import ascii_uppercase
import textwrap

import numpy as np
import pytest

from pandas.compat import PYPY

from pandas import (
    CategoricalIndex,
    MultiIndex,
    Series,
    date_range,
)


def test_info_categorical_column_just_works():
    n = 2500
    data = np.array(list("abcdefghij")).take(
        np.random.default_rng(2).integers(0, 10, size=n, dtype=int)
    )
    s = Series(data).astype("category")
    s.isna()
    buf = StringIO()
    s.info(buf=buf)

    s2 = s[s == "d"]
    buf = StringIO()
    s2.info(buf=buf)


def test_info_categorical():
    # GH14298
    idx = CategoricalIndex(["a", "b"])
    s = Series(np.zeros(2), index=idx)
    buf = StringIO()
    s.info(buf=buf)


@pytest.mark.parametrize("verbose", [True, False])
def test_info_series(lexsorted_two_level_string_multiindex, verbose):
    index = lexsorted_two_level_string_multiindex
    ser = Series(range(len(index)), index=index, name="sth")
    buf = StringIO()
    ser.info(verbose=verbose, buf=buf)
    result = buf.getvalue()

    expected = textwrap.dedent(
        """\
        <class 'pandas.core.series.Series'>
        MultiIndex: 10 entries, ('foo', 'one') to ('qux', 'three')
        """
    )
    if verbose:
        expected += textwrap.dedent(
            """\
            Series name: sth
            Non-Null Count  Dtype
            --------------  -----
            10 non-null     int64
            """
        )
    expected += textwrap.dedent(
        f"""\
        dtypes: int64(1)
        memory usage: {ser.memory_usage()}.0+ bytes
        """
    )
    assert result == expected


def test_info_memory():
    s = Series([1, 2], dtype="i8")
    buf = StringIO()
    s.info(buf=buf)
    result = buf.getvalue()
    memory_bytes = float(s.memory_usage())
    expected = textwrap.dedent(
        f"""\
    <class 'pandas.core.series.Series'>
    RangeIndex: 2 entries, 0 to 1
    Series name: None
    Non-Null Count  Dtype
    --------------  -----
    2 non-null      int64
    dtypes: int64(1)
    memory usage: {memory_bytes} bytes
    """
    )
    assert result == expected


def test_info_wide():
    s = Series(np.random.default_rng(2).standard_normal(101))
    msg = "Argument `max_cols` can only be passed in DataFrame.info, not Series.info"
    with pytest.raises(ValueError, match=msg):
        s.info(max_cols=1)


def test_info_shows_dtypes():
    dtypes = [
        "int64",
        "float64",
        "datetime64[ns]",
        "timedelta64[ns]",
        "complex128",
        "object",
        "bool",
    ]
    n = 10
    for dtype in dtypes:
        s = Series(np.random.default_rng(2).integers(2, size=n).astype(dtype))
        buf = StringIO()
        s.info(buf=buf)
        res = buf.getvalue()
        name = f"{n:d} non-null     {dtype}"
        assert name in res


@pytest.mark.xfail(PYPY, reason="on PyPy deep=True doesn't change result")
def test_info_memory_usage_deep_not_pypy():
    s_with_object_index = Series({"a": [1]}, index=["foo"])
    assert s_with_object_index.memory_usage(
        index=True, deep=True
    ) > s_with_object_index.memory_usage(index=True)

    s_object = Series({"a": ["a"]})
    assert s_object.memory_usage(deep=True) > s_object.memory_usage()


@pytest.mark.xfail(not PYPY, reason="on PyPy deep=True does not change result")
def test_info_memory_usage_deep_pypy():
    s_with_object_index = Series({"a": [1]}, index=["foo"])
    assert s_with_object_index.memory_usage(
        index=True, deep=True
    ) == s_with_object_index.memory_usage(index=True)

    s_object = Series({"a": ["a"]})
    assert s_object.memory_usage(deep=True) == s_object.memory_usage()


@pytest.mark.parametrize(
    "series, plus",
    [
        (Series(1, index=[1, 2, 3]), False),
        (Series(1, index=list("ABC")), True),
        (Series(1, index=MultiIndex.from_product([range(3), range(3)])), False),
        (
            Series(1, index=MultiIndex.from_product([range(3), ["foo", "bar"]])),
            True,
        ),
    ],
)
def test_info_memory_usage_qualified(series, plus):
    buf = StringIO()
    series.info(buf=buf)
    if plus:
        assert "+" in buf.getvalue()
    else:
        assert "+" not in buf.getvalue()


def test_info_memory_usage_bug_on_multiindex():
    # GH 14308
    # memory usage introspection should not materialize .values
    N = 100
    M = len(ascii_uppercase)
    index = MultiIndex.from_product(
        [list(ascii_uppercase), date_range("20160101", periods=N)],
        names=["id", "date"],
    )
    s = Series(np.random.default_rng(2).standard_normal(N * M), index=index)

    unstacked = s.unstack("id")
    assert s.values.nbytes == unstacked.values.nbytes
    assert s.memory_usage(deep=True) > unstacked.memory_usage(deep=True).sum()

    # high upper bound
    diff = unstacked.memory_usage(deep=True).sum() - s.memory_usage(deep=True)
    assert diff < 2000