import pytest def foo(x): return x + 1 def test_foo(): assert foo(1) == 2