diff --git a/5-1.py b/5-1.py
index 3d40511..62463c3 100644
--- a/5-1.py
+++ b/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()