python - 如何使用pytest为斐波那契制作测试类或函数？

Question

def fibR(n):
 if n==1 or n==2:
  return 1
 return fib(n-1)+fib(n-2)
print (fibR(5))

例如，我如何为斐波那契制作测试类？

Answer 1

我将您现有的代码保存在名为 fib.py 的文件中:

def fibR(n):
 if n==1 or n==2:
  return 1
 return fibR(n-1)+fibR(n-2)

在同一目录中，我创建了一个名为 test_fib.py 的文件:

import pytest
from fib import fibR

def test_fib_1_equals_1():
    assert fibR(1) == 1

def test_fib_2_equals_1():
    assert fibR(2) == 1

def test_fib_6_equals_8():
    assert fibR(6) == 8

如果我从命令行在此目录中运行 py.test，我可以使用这些测试自动检查 fibR 的正确性:

collected 3 items 

test_fib.py ...

================= 3 passed in 0.01 seconds ===========