FilesExpand file tree

 main

/

test_conftest.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (30 loc) · 1.58 KB

 main

/

test_conftest.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (30 loc) · 1.58 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

import pytest

from tests import conftest

@pytest.mark.parametrize('function_name', ['assert_index_equal',

'assert_series_equal',

'assert_frame_equal'])

@pytest.mark.parametrize('pd_version', ['1.0.0', '1.1.0'])

@pytest.mark.parametrize('check_less_precise', [True, False])

def test__check_pandas_assert_kwargs(mocker, function_name, pd_version,

check_less_precise):

# test that conftest._check_pandas_assert_kwargs returns appropriate

# kwargs for the assert_x_equal functions

# NOTE: be careful about mixing mocker.patch and pytest.MonkeyPatch!

# they do not coordinate their cleanups, so it is safest to only

# use one or the other. GH #1447

# patch the pandas assert; not interested in actually calling them,

# plus we want to spy on how they get called.

spy = mocker.patch('pandas.testing.' + function_name)

# patch pd.__version__ to exercise the two branches in

# conftest._check_pandas_assert_kwargs

mocker.patch('pandas.__version__', new=pd_version)

# finally, run the function and check what args got passed to pandas:

assert_function = getattr(conftest, function_name)

args = [None, None]

assert_function(*args, check_less_precise=check_less_precise)

if pd_version == '1.1.0':

tol = 1e-3 if check_less_precise else 1e-5

expected_kwargs = {'atol': tol, 'rtol': tol}

else:

expected_kwargs = {'check_less_precise': check_less_precise}

spy.assert_called_once_with(*args, **expected_kwargs)