CS50项目：Twitter功能代码单元测试异常排查求助

问题描述 #

我正在完成CS50的「Testing my Twitter」项目，功能代码运行正常，但单元测试环节遇到问题——现在测试仅输出「OK」，即使修改代码制造错误，结果依然如此。

待测试的功能代码 #

def shorten(word):
    tweet = input("Input: ")
    for letter in tweet:
        if not letter in ['A','E','I','O','U','a','e','i','o']:
            return letter
    print()

单元测试代码 #

import twttr

def test_twttr():
    assert shorten('Hello') == 'Hll'
test_twttr()

问题分析与修复方案 #

核心问题 #

1. 函数未使用传入参数：shorten函数定义了word参数，但完全没用到，反而通过input()读取控制台输入。这导致单元测试传入的'Hello'根本没被处理，断言的结果取决于你手动输入的内容，而非测试用例。
2. 逻辑不符合需求：函数找到第一个非元音字母就直接返回单个字符，比如输入Hello会返回H，而非预期的Hll，完全没实现过滤所有元音并返回剩余字符的功能。
3. 测试「假OK」原因：测试时你手动输入了符合断言的内容，所以断言通过输出「OK」；即使修改代码，只要你输入的内容仍符合断言，结果还是「OK」，本质是测试没真正验证函数逻辑。

修复步骤 #

1. 修正shorten函数逻辑：使用传入的word参数，过滤所有元音后拼接成字符串返回，同时补充遗漏的小写元音u：

def shorten(word):
    vowels = {'A','E','I','O','U','a','e','i','o','u'}
    filtered_chars = [char for char in word if char not in vowels]
    return ''.join(filtered_chars)

2. 验证单元测试：此时运行测试代码，shorten('Hello')会正确返回'Hll'，断言通过；若修改函数制造错误（比如故意返回'H'），会触发AssertionError，测试不再输出「OK」。
3. 清理冗余代码：原函数末尾的print()永远不会被执行（函数已提前return），可直接删除。

内容的提问来源于stack exchange，提问作者Essak Ahmed