FM / util

/usr/local/lib64/python3.6/site-packages/pandas/tests/util/

/usr/local/lib64/python3.6/site-packages/pandas/tests/util

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
conftest.py	476	0644	edit dl rm
test_assert_almost_equal.py	11499	0644	edit dl rm
test_assert_categorical_equal.py	2749	0644	edit dl rm
test_assert_extension_array_equal.py	3471	0644	edit dl rm
test_assert_frame_equal.py	8835	0644	edit dl rm
test_assert_index_equal.py	5664	0644	edit dl rm
test_assert_interval_array_equal.py	2364	0644	edit dl rm
test_assert_numpy_array_equal.py	6361	0644	edit dl rm
test_assert_produces_warning.py	547	0644	edit dl rm
test_assert_series_equal.py	9089	0644	edit dl rm
test_deprecate.py	1626	0644	edit dl rm
test_deprecate_kwarg.py	2043	0644	edit dl rm
test_deprecate_nonkeyword_arguments.py	2713	0644	edit dl rm
test_doc.py	1492	0644	edit dl rm
test_hashing.py	10860	0644	edit dl rm
test_numba.py	308	0644	edit dl rm
test_safe_import.py	1020	0644	edit dl rm
test_show_versions.py	1187	0644	edit dl rm
test_util.py	1982	0644	edit dl rm
test_validate_args.py	1844	0644	edit dl rm
test_validate_args_and_kwargs.py	2391	0644	edit dl rm
test_validate_kwargs.py	1740	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/util/test_assert_extension_array_equal.py (3471B)

import numpy as np
import pytest

from pandas import array
import pandas._testing as tm
from pandas.core.arrays.sparse import SparseArray

@pytest.mark.parametrize(
    "kwargs",
    [
        dict(),  # Default is check_exact=False
        dict(check_exact=False),
        dict(check_exact=True),
    ],
)
def test_assert_extension_array_equal_not_exact(kwargs):
    # see gh-23709
    arr1 = SparseArray([-0.17387645482451206, 0.3414148016424936])
    arr2 = SparseArray([-0.17387645482451206, 0.3414148016424937])

if kwargs.get("check_exact", False):
        msg = """\
ExtensionArray are different

ExtensionArray values are different \\(50\\.0 %\\)
\\[left\\]:  \\[-0\\.17387645482.*, 0\\.341414801642.*\\]
\\[right\\]: \\[-0\\.17387645482.*, 0\\.341414801642.*\\]"""

with pytest.raises(AssertionError, match=msg):
            tm.assert_extension_array_equal(arr1, arr2, **kwargs)
    else:
        tm.assert_extension_array_equal(arr1, arr2, **kwargs)

@pytest.mark.parametrize("decimals", range(10))
def test_assert_extension_array_equal_less_precise(decimals):
    rtol = 0.5 * 10 ** -decimals
    arr1 = SparseArray([0.5, 0.123456])
    arr2 = SparseArray([0.5, 0.123457])

if decimals >= 5:
        msg = """\
ExtensionArray are different

ExtensionArray values are different \\(50\\.0 %\\)
\\[left\\]:  \\[0\\.5, 0\\.123456\\]
\\[right\\]: \\[0\\.5, 0\\.123457\\]"""

with pytest.raises(AssertionError, match=msg):
            tm.assert_extension_array_equal(arr1, arr2, rtol=rtol)
    else:
        tm.assert_extension_array_equal(arr1, arr2, rtol=rtol)

def test_assert_extension_array_equal_dtype_mismatch(check_dtype):
    end = 5
    kwargs = dict(check_dtype=check_dtype)

arr1 = SparseArray(np.arange(end, dtype="int64"))
    arr2 = SparseArray(np.arange(end, dtype="int32"))

if check_dtype:
        msg = """\
ExtensionArray are different

Attribute "dtype" are different
\\[left\\]:  Sparse\\[int64, 0\\]
\\[right\\]: Sparse\\[int32, 0\\]"""

with pytest.raises(AssertionError, match=msg):
            tm.assert_extension_array_equal(arr1, arr2, **kwargs)
    else:
        tm.assert_extension_array_equal(arr1, arr2, **kwargs)

def test_assert_extension_array_equal_missing_values():
    arr1 = SparseArray([np.nan, 1, 2, np.nan])
    arr2 = SparseArray([np.nan, 1, 2, 3])

msg = """\
ExtensionArray NA mask are different

ExtensionArray NA mask values are different \\(25\\.0 %\\)
\\[left\\]:  \\[True, False, False, True\\]
\\[right\\]: \\[True, False, False, False\\]"""

with pytest.raises(AssertionError, match=msg):
        tm.assert_extension_array_equal(arr1, arr2)

@pytest.mark.parametrize("side", ["left", "right"])
def test_assert_extension_array_equal_non_extension_array(side):
    numpy_array = np.arange(5)
    extension_array = SparseArray(numpy_array)

msg = f"{side} is not an ExtensionArray"
    args = (
        (numpy_array, extension_array)
        if side == "left"
        else (extension_array, numpy_array)
    )

with pytest.raises(AssertionError, match=msg):
        tm.assert_extension_array_equal(*args)

@pytest.mark.parametrize("right_dtype", ["Int32", "int64"])
def test_assert_extension_array_equal_ignore_dtype_mismatch(right_dtype):
    # https://github.com/pandas-dev/pandas/issues/35715
    left = array([1, 2, 3], dtype="Int64")
    right = array([1, 2, 3], dtype=right_dtype)
    tm.assert_extension_array_equal(left, right, check_dtype=False)

import numpy as np import pytest from pandas import array import pandas._testing as tm from pandas.core.arrays.sparse import SparseArray @pytest.mark.parametrize( "kwargs", [ dict(), # Default is check_exact=False dict(check_exact=False), dict(check_exact=True), ], ) def test_assert_extension_array_equal_not_exact(kwargs): # see gh-23709 arr1 = SparseArray([-0.17387645482451206, 0.3414148016424936]) arr2 = SparseArray([-0.17387645482451206, 0.3414148016424937]) if kwargs.get("check_exact", False): msg = """\ ExtensionArray are different ExtensionArray values are different \\(50\\.0 %\\) \\[left\\]: \\[-0\\.17387645482.*, 0\\.341414801642.*\\] \\[right\\]: \\[-0\\.17387645482.*, 0\\.341414801642.*\\]""" with pytest.raises(AssertionError, match=msg): tm.assert_extension_array_equal(arr1, arr2, **kwargs) else: tm.assert_extension_array_equal(arr1, arr2, **kwargs) @pytest.mark.parametrize("decimals", range(10)) def test_assert_extension_array_equal_less_precise(decimals): rtol = 0.5 * 10 ** -decimals arr1 = SparseArray([0.5, 0.123456]) arr2 = SparseArray([0.5, 0.123457]) if decimals >= 5: msg = """\ ExtensionArray are different ExtensionArray values are different \\(50\\.0 %\\) \\[left\\]: \\[0\\.5, 0\\.123456\\] \\[right\\]: \\[0\\.5, 0\\.123457\\]""" with pytest.raises(AssertionError, match=msg): tm.assert_extension_array_equal(arr1, arr2, rtol=rtol) else: tm.assert_extension_array_equal(arr1, arr2, rtol=rtol) def test_assert_extension_array_equal_dtype_mismatch(check_dtype): end = 5 kwargs = dict(check_dtype=check_dtype) arr1 = SparseArray(np.arange(end, dtype="int64")) arr2 = SparseArray(np.arange(end, dtype="int32")) if check_dtype: msg = """\ ExtensionArray are different Attribute "dtype" are different \\[left\\]: Sparse\\[int64, 0\\] \\[right\\]: Sparse\\[int32, 0\\]""" with pytest.raises(AssertionError, match=msg): tm.assert_extension_array_equal(arr1, arr2, **kwargs) else: tm.assert_extension_array_equal(arr1, arr2, **kwargs) def test_assert_extension_array_equal_missing_values(): arr1 = SparseArray([np.nan, 1, 2, np.nan]) arr2 = SparseArray([np.nan, 1, 2, 3]) msg = """\ ExtensionArray NA mask are different ExtensionArray NA mask values are different \\(25\\.0 %\\) \\[left\\]: \\[True, False, False, True\\] \\[right\\]: \\[True, False, False, False\\]""" with pytest.raises(AssertionError, match=msg): tm.assert_extension_array_equal(arr1, arr2) @pytest.mark.parametrize("side", ["left", "right"]) def test_assert_extension_array_equal_non_extension_array(side): numpy_array = np.arange(5) extension_array = SparseArray(numpy_array) msg = f"{side} is not an ExtensionArray" args = ( (numpy_array, extension_array) if side == "left" else (extension_array, numpy_array) ) with pytest.raises(AssertionError, match=msg): tm.assert_extension_array_equal(*args) @pytest.mark.parametrize("right_dtype", ["Int32", "int64"]) def test_assert_extension_array_equal_ignore_dtype_mismatch(right_dtype): # https://github.com/pandas-dev/pandas/issues/35715 left = array([1, 2, 3], dtype="Int64") right = array([1, 2, 3], dtype=right_dtype) tm.assert_extension_array_equal(left, right, check_dtype=False)