编写Python小程序来统计测试脚本的关键字

编写Python小程序来统计测试脚本的关键字的攻略如下：

确认需求

首先需要明确需求，统计测试脚本的关键字，需要考虑以下几个方面：

1. 如何读取测试脚本文件？
2. 如何解析测试脚本内容获取关键字？
3. 如何将关键字进行统计并输出结果？

明确需求后，我们就可以开始编写代码了。

编写Python程序

读取文件

读取文件的代码如下：

with open("test_script.py", "r") as f:
    content = f.read()

上述代码中，使用with open语句打开文件，将文件内容读取到变量content中。

解析内容

获取测试脚本中的关键字，可以使用Python中的正则表达式进行匹配。例如，假设需要匹配以下格式的关键字：

# test_case_1
def test_case_1():
    # 打开浏览器
    open_browser()
    # 输入url
    input_url("https://www.example.com")
    # 点击登录按钮
    click_button(".login_button")
    # 断言登录是否成功
    assert_element_exist(".user_name")

针对上述格式，可以使用如下正则表达式进行匹配：

pattern = re.compile(r'#\s*(\w+)\s+def\s+(\w+)\(')
matches = re.findall(pattern, content)

上述代码使用了re.compile将正则表达式编译成一个模式，然后使用re.findall函数匹配所有符合条件的关键字。其中(\w+)表示匹配一个或多个字母或数字，(\w+)\(表示匹配匹配一个或多个字母或数字后跟一个左括号。

统计关键字

获取到关键字后，可以将其进行统计。例如，可以使用Python的collections模块中的 defaultdict 类来实现计数器：

from collections import defaultdict

counter = defaultdict(int)
for match in matches:
    counter[match[0]] += 1

上述代码使用defaultdict(int)创建一个默认值为0的计数器，遍历正则匹配后的结果matches，将关键字作为计数器的键，出现次数作为值，累加到计数器中。

输出结果

最后，将统计结果进行输出。例如，可以使用如下代码将结果输出到文件：

with open("result.txt", "w") as f:
    for key, value in counter.items():
        f.write(f"{key}: {value}\n")

上述代码使用with open语句进行文件的写入操作，并使用counter.items()遍历计数器中的键值对，将其写入文件。

示例

以下是一个完整的例子：

test_script.py

# test_case_1
def test_case_1():
    # 打开浏览器
    open_browser()
    # 输入url
    input_url("https://www.example.com")
    # 点击登录按钮
    click_button(".login_button")
    # 断言登录是否成功
    assert_element_exist(".user_name")

# test_case_2
def test_case_2():
    # 打开浏览器
    open_browser()
    # 输入url
    input_url("https://www.example.com")
    # 点击注册按钮
    click_button(".register_button")
    # 断言注册是否成功
    assert_element_exist(".user_name")

统计程序代码

import re
from collections import defaultdict

with open("test_script.py", "r") as f:
    content = f.read()

pattern = re.compile(r'#\s*(\w+)\s+def\s+(\w+)\(')
matches = re.findall(pattern, content)

counter = defaultdict(int)
for match in matches:
    counter[match[0]] += 1

with open("result.txt", "w") as f:
    for key, value in counter.items():
        f.write(f"{key}: {value}\n")

输出结果

假设以上代码保存在count_keywords.py文件中，使用如下命令执行程序并输出结果：

$ python count_keywords.py

结果将输出到result.txt文件中，内容如下：

test_case_1: 1
test_case_2: 1