在pycharm里面使用pytest.assume(True) ，assume高亮显示，但是运行代码没有报错

[Wc-lab]

[abeelan]

当我的 pytest=5.4.3 时，遇到了和你一样的问题。

更新版本为 pytest 7.2.0 后，问题解决。

[Dragon1573]

Envrionment

Runtime Envrionments:

pwsh> python --version       
Python 3.12.1

pwsh> pipx --version
1.4.3

pwsh> pdm --version    
PDM, version 2.12.2

pwsh> pdm list --tree '*pytest*'
pytest-assume 2.4.3 [ required: >=2.4.3 ]
├── pytest 7.4.4 [ required: >=2.7 ]
│   ├── colorama 0.4.6 [ required: Any ]
│   ├── iniconfig 2.0.0 [ required: Any ]
│   ├── packaging 23.2 [ required: Any ]
│   └── pluggy 1.4.0 [ required: <2.0,>=0.12 ]
└── six 1.16.0 [ required: Any ]

Visual Studio Code:

版本: 1.85.2 (system setup)
提交: 8b3775030ed1a69b13e4f4c628c612102e30a681
日期: 2024-01-18T06:40:10.514Z
Electron: 25.9.7
ElectronBuildId: 26354273
Chromium: 114.0.5735.289
Node.js: 18.15.0
V8: 11.4.183.29-electron.0
OS: Windows_NT x64 10.0.22635

Description

When trying from pytest import assume, Pylance is complaining because there's no assume in pytest, and regard assume with type object.

When trying the following snippets, Pylance will also complain about it.

# An object of type "object" cannot be used with "with" because it has not implemented `__enter__`. Pylance (reportGeneralTypeIssues)
# An object of type "object" cannot be used with "with" because it has not implemented `__exit__`. Pylance (reportGeneralTypeIssues)
with pytest.assume:
    assert True

Additional Informations

No errors occured because this problem when running with pdm run pytest. AssumptionFailure is raised as expected from assume plugin in pytest.

============================================================================= FAILURES ==============================================================================
__________________________________________________________________________ test_regression __________________________________________________________________________

    def test_regression():
        from pypynum import regression

        with pytest.assume:
            assert regression.linear_regression([0, 1, 2, 3, 4], [2, 4, 6, 7, 8]) == [1.5, 2.4]
        with pytest.assume:
            assert regression.parabolic_regression([0, 1, 2, 3, 4], [2, 4, 6, 7, 8]) == [
                -0.214285714,
                2.357142857,
                1.971428571,
            ]
        with pytest.assume:
>           assert regression.polynomial_regression([0, 1, 2, 3, 4], [2, 4, 6, 7, 8], 4) == False
E           pytest_assume.plugin.FailedAssumption: 
E           3 Failed Assumptions:
E
E           test\test_all.py:217: AssumptionFailure
E           >>  with pytest.assume:
E           AssertionError: assert [1.5, 2.4000000000000004] == [1.5, 2.4]
E             At index 1 diff: 2.4000000000000004 != 2.4
E             Full diff:
E             - [1.5, 2.4]
E             + [1.5, 2.4000000000000004]
E
E           test\test_all.py:219: AssumptionFailure
E           >>  with pytest.assume:
E           AssertionError: assert [-0.214285714...1428571428569] == [-0.214285714..., 1.971428571]
E             At index 0 diff: -0.21428571428571563 != -0.214285714
E             Full diff:
E             - [-0.214285714, 2.357142857, 1.971428571]
E             + [-0.21428571428571563, 2.3571428571428625, 1.971428571428569]
E             ?              ++++++++             +++++++             ++++++
E
E           test\test_all.py:225: AssumptionFailure
E           >>  with pytest.assume:
E           AssertionError: assert [0.0833333333480164, -0.6666666668091551, 1.416666667838451, 1.1666666648311779, 2.0000000002900586] == False
E            +  where [0.0833333333480164, -0.6666666668091551, 1.416666667838451, 1.1666666648311779, 2.0000000002900586] = <function polynomial_regression at 0x000001E6AAFD6D40>([0, 1, 2, 3, 4], [2, 4, 6, 7, 8], 4)
E            +    where <function polynomial_regression at 0x000001E6AAFD6D40> = <module 'pypynum.regression' from 'D:\\Repository\\PyPyNum\\src\\pypynum\\regression.py'>.polynomial_regression

test\test_all.py:226: FailedAssumption
========================================================================= warnings summary ========================================================================== 
test/test_all.py::test_regression
  D:\Repository\PyPyNum\.venv\Lib\site-packages\_pytest\runner.py:262: PluggyTeardownRaisedWarning: A plugin raised an exception during an old-style hookwrapper teardown.
  Plugin: assume, Hook: pytest_runtest_call
  FailedAssumption:
  3 Failed Assumptions:

  test\test_all.py:217: AssumptionFailure
  >>    with pytest.assume:
  AssertionError: assert [1.5, 2.4000000000000004] == [1.5, 2.4]
    At index 1 diff: 2.4000000000000004 != 2.4
    Full diff:
    - [1.5, 2.4]
    + [1.5, 2.4000000000000004]

  test\test_all.py:219: AssumptionFailure
  >>    with pytest.assume:
  AssertionError: assert [-0.214285714...1428571428569] == [-0.214285714..., 1.971428571]
    At index 0 diff: -0.21428571428571563 != -0.214285714
    Full diff:
    - [-0.214285714, 2.357142857, 1.971428571]
    + [-0.21428571428571563, 2.3571428571428625, 1.971428571428569]
    ?              ++++++++             +++++++             ++++++

  test\test_all.py:225: AssumptionFailure
  >>    with pytest.assume:
  AssertionError: assert [0.0833333333480164, -0.6666666668091551, 1.416666667838451, 1.1666666648311779, 2.0000000002900586] == False
   +  where [0.0833333333480164, -0.6666666668091551, 1.416666667838451, 1.1666666648311779, 2.0000000002900586] = <function polynomial_regression at 0x000001E6AAFD6D40>([0, 1, 2, 3, 4], [2, 4, 6, 7, 8], 4)
   +    where <function polynomial_regression at 0x000001E6AAFD6D40> = <module 'pypynum.regression' from 'D:\\Repository\\PyPyNum\\src\\pypynum\\regression.py'>.polynomial_regression


  For more information see https://pluggy.readthedocs.io/en/stable/api_reference.html#pluggy.PluggyTeardownRaisedWarning
    lambda: ihook(item=item, **kwds), when=when, reraise=reraise

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
====================================================================== short test summary info ====================================================================== 
FAILED test/test_all.py::test_regression - pytest_assume.plugin.FailedAssumption:
============================================================== 1 failed, 16 passed, 1 warning in 0.30s ============================================================== 

[Dragon1573]

Problem solved ❤️

Just change from pytest import assume to from pytest_assume.plugin import assume, then Pylance in Visual Studio Code can inspect what assume acutally is and provide full functionality of auto-completions.

[Sanofi-EL]

If I do this:

from pytest_assume.plugin import assume

Then I get:

Stub file not found for "pytest_assume.plugin"PylancereportMissingTypeStubs
(module) pytest_assume

As for if I import simply:

import pytest

Then at pytest.assume:

Type of "assume" is unknownPylancereportUnknownMemberType

I am guessing that this is more of a Pylance issue than pytest_assume.