在我们编写完自定义模块以后,最好在模块中对代码进行提前测试,以防止有任何异常。
测试需求:
① 要求我们可以直接在模块文件中对代码进行直接测试
② 代码测试完毕后,当导入到其他文件时,测试代码要自动失效
引入一个魔术变量:__name__
,其保存的内容就是一个==字符串类型==的数据。
==随着运行页面的不同,其返回结果也是不同的:==
① 如果__name__
是在当前页面运行时,其返回结果为__main__
② 如果__name__
在第三方页面导入运行时,其返回结果为模块名称(文件名称)
基于以上特性,我们可以把__name__
编写在自定义模块中,其语法如下:
if __name__ == '__main__':
# 执行测试代码
以上代码主要出现在自定义模块中,主要用于实现代码测试
__name__
魔术方法除了可以在自定义模块中测试使用,还可以用于编写程序的入口:
Python程序一般都是从入口开始执行的:
# 定义一个main方法(入口文件)
def main():
# 执行我们要执行的功能
# ① 打印选择菜单
# ② 添加学员信息
# ③ 删除学员信息
# ...
# 调用执行入口
if __name__ == '__main__':
main()