test(5-1.py):优化相似度测试逻辑并增强错误提示

- 打印相似度测试结果及其数量,提供更详细的输出信息
- 增加对无匹配记录的警告提示,提高错误处理的友好性
- 保留原有的断言测试,确保代码功能不变
This commit is contained in:
fly6516 2025-04-20 03:09:53 +08:00
parent dcf2d79bbf
commit 14552b200f

8
5-1.py
View File

@ -159,10 +159,14 @@ print("Number of similarity records: {}".format(similaritiesFullRDD.count()))
# 计算并测试相似度
similarity_test = similaritiesFullRDD.filter(lambda x: x[0][0] == 'b00005lzly' and x[0][1] == 'http://www.google.com/base/feeds/snippets/13823221823254120257').collect()
print(len(similarity_test))
print("Similarity test results:", similarity_test)
print("Number of similarity test results:", len(similarity_test))
# 测试
assert len(similarity_test) >= 1, "incorrect len(similarity_test)"
if len(similarity_test) == 0:
print("Warning: No matching records found for the given IDs.")
else:
assert len(similarity_test) >= 1, "incorrect len(similarity_test)"
assert similaritiesFullRDD.count() == 27411, "incorrect similaritiesFullRDD.count()"
sc.stop()