Beta Shell v2.0

File Manager

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

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	0 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:40:26	Del
accessors	-	drwxr-xr-x	2024-04-12 10:40:25	Del
indexing	-	drwxr-xr-x	2024-04-12 10:40:25	Del
methods	-	drwxr-xr-x	2024-04-12 10:40:25	Del
test_api.py	10.1 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_arithmetic.py	33.02 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_constructors.py	83.25 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_cumulative.py	4.92 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_formats.py	16.72 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_iteration.py	1.38 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_logical_ops.py	19.73 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_missing.py	3.2 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_npfuncs.py	1.07 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_reductions.py	6.37 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_subclass.py	2.6 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_ufunc.py	14.35 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_unary.py	1.58 KB	-rw-r--r--	2024-04-12 10:40:17	Edit Del
test_validate.py	668 B	-rw-r--r--	2024-04-12 10:40:17	Edit Del

Editing: test_unary.py (1.58 KB)

import pytest

from pandas import Series
import pandas._testing as tm


class TestSeriesUnaryOps:
    # __neg__, __pos__, __invert__

    def test_neg(self):
        ser = Series(range(5), dtype="float64", name="series")
        tm.assert_series_equal(-ser, -1 * ser)

    def test_invert(self):
        ser = Series(range(5), dtype="float64", name="series")
        tm.assert_series_equal(-(ser < 0), ~(ser < 0))

    @pytest.mark.parametrize(
        "source, neg_target, abs_target",
        [
            ([1, 2, 3], [-1, -2, -3], [1, 2, 3]),
            ([1, 2, None], [-1, -2, None], [1, 2, None]),
        ],
    )
    def test_all_numeric_unary_operators(
        self, any_numeric_ea_dtype, source, neg_target, abs_target
    ):
        # GH38794
        dtype = any_numeric_ea_dtype
        ser = Series(source, dtype=dtype)
        neg_result, pos_result, abs_result = -ser, +ser, abs(ser)
        if dtype.startswith("U"):
            neg_target = -Series(source, dtype=dtype)
        else:
            neg_target = Series(neg_target, dtype=dtype)

        abs_target = Series(abs_target, dtype=dtype)

        tm.assert_series_equal(neg_result, neg_target)
        tm.assert_series_equal(pos_result, ser)
        tm.assert_series_equal(abs_result, abs_target)

    @pytest.mark.parametrize("op", ["__neg__", "__abs__"])
    def test_unary_float_op_mask(self, float_ea_dtype, op):
        dtype = float_ea_dtype
        ser = Series([1.1, 2.2, 3.3], dtype=dtype)
        result = getattr(ser, op)()
        target = result.copy(deep=True)
        ser[0] = None
        tm.assert_series_equal(result, target)