单元测试以在len（X）-len（pred_values）！= num_step时提醒我

Question

我对“ unittest”主题不太满意。我想创建一个单元测试，以便说“嘿，这是错误的（或正确的）答案，因为blabla！”。我需要进行一次单元测试，因为花了3 MF个星期才能找到预测为何无效的原因！因此，我希望将来避免此类错误。

问题：

1. 当len(X) - len(pred_values)不等于num_step时，如何要求代码提醒我？
2. 我是否需要创建一个单元测试文件来收集所有单元测试，例如unittest.py？
3. 我们是否需要将单元测试与主要代码分开放置？

Answer 1

Pytest非常适合单元测试。它背后有强大的社区，大量的文档以及数量众多的库，可简化单元测试的寿命。

pip install pytest

创建一个名为tests的文件夹。在其中添加__init__.py和一个名为test_thing.py的文件，然后添加：

def test_foo():
    assert len(X) - len(pred_values) ==num_steps

然后运行pytest .