U
|
®ý°dÉ ã @ s d Z ddlmZ ddlZddlmZ ddlZddlZddl m
|
Z
|
ddl m Z ddl mZmZ ddlmZ dd lmZmZ dd
|
lmZ d9d d d ddZddddZddddZddddZd dddZd:d d dd d Z ej j e d d Z ej j e d d Z ej j e d!d Z!ej j e d"d Z"ej j e d#e #¡ d d Z$ej j e d$d Z%ej j e d%e d&e d Z&dd d d d'd(d)Z'd*dd+d,Z(d-d. Z)d/dd0d1Z*ej j+e
|
d2d3kd4d Z,ej j e
|
d2d3kd5d Z-ej j+e
|
d6d7d Z.ej j e
|
d6d8d Z/dS );aÂ
|
This module provides decorator functions which can be applied to test objects
|
in order to skip those objects when certain conditions occur. A sample use case
|
is to detect if the platform is missing ``matplotlib``. If so, any test objects
|
which require ``matplotlib`` and decorated with ``@td.skip_if_no_mpl`` will be
|
skipped by ``pytest`` during the execution of the test suite.
|
|
To illustrate, after importing this module:
|
|
import pandas.util._test_decorators as td
|
|
The decorators can be applied to classes:
|
|
@td.skip_if_some_reason
|
class Foo:
|
...
|
|
Or individual functions:
|
|
@td.skip_if_some_reason
|
def test_foo():
|
...
|
|
For more information, refer to the ``pytest`` documentation on ``skipif``.
|
é )Ú annotationsN)ÚCallable)Ú
|
get_option)ÚF)ÚIS64Úis_platform_windows)Úimport_optional_dependency)ÚNUMEXPR_INSTALLEDÚ USE_NUMEXPR)ÚVersionÚstrz
|
str | None)Úmod_nameÚ min_versionc C s¬ z t | }W n8 tk
|
r" Y dS tk
|
rD | dkr>Y dS Y nX |sN|S ddl}zt|j| d}W n$ tk
|
r t|j| d}Y nX |r¨t|t|kr¨|S dS )a
|
Parameters
|
----------
|
mod_name : str
|
Name of the module to be imported
|
min_version : str, default None
|
Minimum required version of the specified mod_name
|
|
Returns
|
-------
|
object
|
The imported module if successful, or False
|
FZnumbar NÚ __version__Z __VERSION__)Ú
|
__import__Ú ImportErrorÚ SystemErrorÚsysÚgetattrÚmodulesÚAttributeErrorr )r r Úmodr Úversion© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/util/_test_decorators.pyÚ safe_import2 s$ r Úbool)Úreturnc C s" t d} | r| d¡ dS dS d S )NZ
|
matplotlibZAggFT)r Zuse)r r r r Ú_skip_if_no_mpl\ s
|
|
r c C s t ¡ \} }| dkrdS dS )NÚen_USTF)ÚlocaleÚ getlocale)ÚlangÚ_r r r Ú_skip_if_not_us_localee s r$ c C s" t do t do t do t d S )Nz scipy.statsz scipy.sparsezscipy.interpolatez scipy.signal)r r r r r Ú_skip_if_no_scipyl s ÿþür% ©Úpackagec C s t jjt| d| ddS )z
|
Skip a test if a package is installed.
|
|
Parameters
|
----------
|
package : str
|
The name of the package.
|
zSkipping because z is installed.©Úreason©ÚpytestÚmarkÚskipifr r& r r r Úskip_if_installedw s
|
ÿr. )r' r c C s8 d| d}|r |d| 7 }t jjt| |d |dS )aù
|
Generic function to help skip tests when required packages are not
|
present on the testing system.
|
|
This function returns a pytest mark with a skip condition that will be
|
evaluated during test collection. An attempt will be made to import the
|
specified ``package`` and optionally ensure it meets the ``min_version``
|
|
The mark can be used as either a decorator for a test function or to be
|
applied to parameters in pytest.mark.parametrize calls or parametrized
|
fixtures.
|
|
If the import and version check are unsuccessful, then the test function
|
(or test case when used in conjunction with parametrization) will be
|
skipped.
|
|
Parameters
|
----------
|
package: str
|
The name of the required package.
|
min_version: str or None, default None
|
Optional minimum version of the package.
|
|
Returns
|
-------
|
_pytest.mark.structures.MarkDecorator
|
a pytest.mark.skipif to use as either a test decorator or a
|
parametrization mark.
|
zCould not import 'ú'z satisfying a min_version of )r r( r* )r' r Úmsgr r r Ú
|
skip_if_no s ÿr1 z Missing matplotlib dependencyr( zmatplotlib is presentzskipping for 32 bitzRunning on WindowszSpecific locale is set zMissing SciPy requirementznumexpr enabled->z , installed->)Úver_strr) c G s: |d krd| d}t jjttjt| k f|d|iS )NzNumPy z or greater requiredr) )r+ r, r- r Únpr )r2 r) Úargsr r r Ú skip_if_np_lt s ÿþýr5 zCallable[[F], F]c s fdd}|S )a
|
Intended for use as a decorator for parametrized fixture,
|
this function will wrap the decorated function with a pytest
|
``parametrize_fixture_doc`` mark. That mark will format
|
initial fixture docstring by replacing placeholders {0}, {1} etc
|
with parameters passed as arguments.
|
|
Parameters
|
----------
|
args: iterable
|
Positional arguments for docstring.
|
|
Returns
|
-------
|
function
|
The decorated function wrapped within a pytest
|
``parametrize_fixture_doc`` mark
|
c s | j j | _ | S )N)Ú__doc__Úformat)Zfixture©r4 r r Údocumented_fixtureà s z3parametrize_fixture_doc.<locals>.documented_fixturer )r4 r9 r r8 r Úparametrize_fixture_docÌ s r: c C s< zt d tjj} W n" tk
|
r6 tjjdd} Y nX | S )NZpytest_asyncioz!Missing dependency pytest-asyncior( )r r+ r, Zasyncior Úskip)Ú
|
async_markr r r r< ç s r< ÚNonec C s t jjdd}| j |¡ d S )Nú$Not yet implemented for ArrayManagerr( )r+ r, ÚxfailÚnodeZ
|
add_marker)Úrequestr, r r r Ú&mark_array_manager_not_yet_implementedñ s rB zmode.data_managerÚarrayr> z@Test that relies on BlockManager internals or specific behaviourzmode.copy_on_writez2Not yet implemented/adapted for Copy-on-Write modez%Test not valid for Copy-on-Write mode)N)N)0r6 Ú
|
__future__r r Útypingr Únumpyr3 r+ Zpandas._configr Zpandas._typingr Z pandas.compatr r Zpandas.compat._optionalr Z#pandas.core.computation.expressionsr r
|
Zpandas.util.versionr r r r$ r% r. r1 r, r- Zskip_if_no_mplZ skip_if_mplZ skip_if_32bitZskip_if_windowsr! Zskip_if_not_us_localeZskip_if_no_scipyZ skip_if_no_ner5 r: r< rB r? Z&skip_array_manager_not_yet_implementedZ skip_array_manager_invalid_testZ&skip_copy_on_write_not_yet_implementedZ skip_copy_on_write_invalid_testr r r r Ú<module> sp * & ÿþ ÿþ
|
|
|
þ
|
þþþ
|
