Beta Shell v2.0

File Manager

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

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	0 B	-rw-r--r--	2024-04-12 10:40:16	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:40:20	Del
common.py	4.26 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
conftest.py	3.39 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_array_ops.py	1.04 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_categorical.py	742 B	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_datetime64.py	88.14 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_interval.py	10.69 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_numeric.py	54.21 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_object.py	13.21 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_period.py	58.22 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del
test_timedelta64.py	76.94 KB	-rw-r--r--	2024-04-12 10:40:16	Edit Del

Editing: test_array_ops.py (1.04 KB)

import operator

import numpy as np
import pytest

import pandas._testing as tm
from pandas.core.ops.array_ops import (
    comparison_op,
    na_logical_op,
)


def test_na_logical_op_2d():
    left = np.arange(8).reshape(4, 2)
    right = left.astype(object)
    right[0, 0] = np.nan

    # Check that we fall back to the vec_binop branch
    with pytest.raises(TypeError, match="unsupported operand type"):
        operator.or_(left, right)

    result = na_logical_op(left, right, operator.or_)
    expected = right
    tm.assert_numpy_array_equal(result, expected)


def test_object_comparison_2d():
    left = np.arange(9).reshape(3, 3).astype(object)
    right = left.T

    result = comparison_op(left, right, operator.eq)
    expected = np.eye(3).astype(bool)
    tm.assert_numpy_array_equal(result, expected)

    # Ensure that cython doesn't raise on non-writeable arg, which
    #  we can get from np.broadcast_to
    right.flags.writeable = False
    result = comparison_op(left, right, operator.ne)
    tm.assert_numpy_array_equal(result, ~expected)