test(5-1.py):优化相似度测试逻辑并增强错误提示
- 打印相似度测试结果及其数量,提供更详细的输出信息 - 增加对无匹配记录的警告提示,提高错误处理的友好性 - 保留原有的断言测试,确保代码功能不变
This commit is contained in:
parent
dcf2d79bbf
commit
14552b200f
8
5-1.py
8
5-1.py
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user