diff options
author | Anthony Sottile <asottile@umich.edu> | 2020-05-07 13:14:58 -0700 |
---|---|---|
committer | Anthony Sottile <asottile@umich.edu> | 2020-05-07 13:14:58 -0700 |
commit | b238845d0f4935f805521b09073211e50251f0dc (patch) | |
tree | 8a2e3f3529a3e3c3ff5bade4f0ec41d988a35f4b /testing/test_doctest.py | |
parent | de556f895febd89d14db0a0828e5c8555c75f44e (diff) | |
download | pytest-b238845d0f4935f805521b09073211e50251f0dc.tar.gz |
Fix _is_setup_py for files encoded differently than locale
Diffstat (limited to 'testing/test_doctest.py')
-rw-r--r-- | testing/test_doctest.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/testing/test_doctest.py b/testing/test_doctest.py index c9defec5d..39afb4e98 100644 --- a/testing/test_doctest.py +++ b/testing/test_doctest.py @@ -5,6 +5,7 @@ import pytest from _pytest.compat import MODULE_NOT_FOUND_ERROR from _pytest.doctest import _get_checker from _pytest.doctest import _is_mocked +from _pytest.doctest import _is_setup_py from _pytest.doctest import _patch_unwrap_mock_aware from _pytest.doctest import DoctestItem from _pytest.doctest import DoctestModule @@ -1487,3 +1488,27 @@ def test_warning_on_unwrap_of_broken_object(stop): with pytest.raises(KeyError): inspect.unwrap(bad_instance, stop=stop) assert inspect.unwrap.__module__ == "inspect" + + +def test_is_setup_py_not_named_setup_py(tmpdir): + not_setup_py = tmpdir.join("not_setup.py") + not_setup_py.write('from setuptools import setup; setup(name="foo")') + assert not _is_setup_py(not_setup_py) + + +@pytest.mark.parametrize("mod", ("setuptools", "distutils.core")) +def test_is_setup_py_is_a_setup_py(tmpdir, mod): + setup_py = tmpdir.join("setup.py") + setup_py.write('from {} import setup; setup(name="foo")'.format(mod)) + assert _is_setup_py(setup_py) + + +@pytest.mark.parametrize("mod", ("setuptools", "distutils.core")) +def test_is_setup_py_different_encoding(tmpdir, mod): + setup_py = tmpdir.join("setup.py") + contents = ( + "# -*- coding: cp1252 -*-\n" + 'from {} import setup; setup(name="foo", description="€")\n'.format(mod) + ) + setup_py.write_binary(contents.encode("cp1252")) + assert _is_setup_py(setup_py) |