### 创建自定义模块并在其他源文件中调用和测试 #### 创建自定义模块 在 Python 中，任何 `.py` 文件都可以作为一个模块被创建。假设有一个名为 `mymodule.py` 的文件，其中包含了若干函数或类的定义。 ```python # mymodule.py def greet(name): """打印问候语""" return f"Hello, {name}!" class Calculator: """简单的计算器类""" def add(self, a, b): return a + b def subtract(self, a, b): return a - b ``` 这段代码展示了如何在一个单独的文件中定义一个函数 `greet` 和一个类 `Calculator`[^1]。 #### 将模块放置在可访问的位置 为了让 Python 找到这个模块，需要将其放在当前工作目录下，或者添加其所在路径至 `sys.path` 列表中。默认情况下，Python 会在脚本所在的目录以及其他标准库路径中查找模块。 如果希望永久扩展模块搜索路径，则可以通过修改环境变量 PYTHONPATH 或者将模块安装到站点包（site-packages）中实现。 #### 在另一个文件中调用模块 要从另一个 Python 文件中使用刚才创建的模块，可以利用 `import` 关键字加载它。例如，在同一项目下的另一文件 `main.py` 中这样写： ```python # main.py import mymodule print(mymodule.greet("Alice")) # 输出: Hello, Alice! calc_instance = mymodule.Calculator() result_addition = calc_instance.add(5, 3) result_subtraction = calc_instance.subtract(5, 3) print(f"Addition Result: {result_addition}") # 输出: Addition Result: 8 print(f"Subtraction Result: {result_subtraction}") # 输出: Subtraction Result: 2 ``` 这里演示了两种方式——直接引用函数以及实例化类成员来进行操作。 #### 测试已创建的模块 对于小型应用来说，可以在同一个主程序里验证功能；但对于较大规模的应用开发而言，建议分离单元测试逻辑以便维护清晰度。unittest 是内置的标准框架之一，可用于构建自动化测试套件。 下面是一个针对前面提到的 `mymodule` 的简单测试例子： ```python # test_mymodule.py import unittest from mymodule import Calculator, greet class TestMyModule(unittest.TestCase): def setUp(self): self.calc = Calculator() def test_greeting_message(self): message = greet('Bob') self.assertEqual(message, 'Hello, Bob!') def test_calculator_operations(self): addition_result = self.calc.add(7, 6) subtraction_result = self.calc.subtract(9, 4) self.assertEqual(addition_result, 13) self.assertEqual(subtraction_result, 5) if __name__ == '__main__': unittest.main() ``` 运行此测试文件将会自动执行所有的断言检查，并报告是否有失败的情况发生。 ---