通过一个简单的例子学习TDD

TDD 学习示例：通过一个简单的例子学习TDD

在学习 TDD（测试驱动开发）时，我们将通过一个简单的例子来了解如何编写测试用例，并基于代码的单元测试来开发。在这个例子中，我们将构建一个简单的 Web 应用程序，该应用程序允许用户搜索并查看一些动物的信息。

测试驱动开发 (TDD)

TDD 是一种编程方法，要求先编写测试用例，然后编写实现，最后再对代码进行重构。这种方法有助于确保代码符合预期，并且易于维护。

测试用例

在编写测试用例时，我们需要考虑如何对代码进行测试。在这个例子中，我们将编写一个测试用例，用于验证从应用程序中获取动物信息，并检查返回的动物信息是否正确。

def test_get_animal_info():
    animal_info = get_animal_info("Bear")
    assert animal_info["name"] == "Bear"
    assert animal_info["classification"] == "Mammal"
    assert animal_info["habitat"] == "Forest"

在这个测试用例中，我们通过调用 get_animal_info 函数并传入 "Bear" 作为参数，来获取动物的详细信息。然后，我们使用 assert 语句来验证返回的动物信息是否正确。

实现

在编写实现时，我们需要根据测试用例来编写代码。在这个例子中，我们需要实现 get_animal_info 函数，该函数将根据动物的名称来获取动物的详细信息。

def get_animal_info(animal_name):
    animal_info = {"name": animal_name, "classification": "Mammal", "habitat": "Forest"}
    return animal_info

在这个实现中，我们使用一个字典来存储动物的详细信息，并将其作为函数的返回值。然后，我们使用 assert 语句来验证返回的动物信息是否正确。

重构

在重构代码时，我们需要确保代码的可读性和可维护性。在这个例子中，我们可以考虑重构代码，以使其更易于理解和维护。

def get_animal_info(animal_name):
    animal_info = {"name": animal_name, "classification": "Mammal", "habitat": "Forest"}
    return animal_info

def get_animal_classification(animal_info):
    classification = animal_info.get("classification")
    if classification == "Mammal":
        return "Mammal"
    elif classification == "Bird":
        return "Bird"
    else:
        return "Other"

def get_animal_habitat(animal_info):
    habitat = animal_info.get("habitat")
    if habitat == "Forest":
        return "Forest"
    elif habitat == "Water":
        return "Water"
    else:
        return "Other"

在这个重构后的代码中，我们使用 get 方法来获取字典中的值，这样代码更易于理解和维护。我们还定义了两个新的函数 get_animal_classification 和 get_animal_habitat，用于获取动物的详细分类和栖息地信息。