Skip to content

Commit 0b6e320

Browse files
committed
fix: add missing category assertions in bot tests
Address PR review: add $ai_bot_category assertions for Bytespider, CCBot, Applebot-Extended, Meta-ExternalAgent.
1 parent 01a3902 commit 0b6e320

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test_ai_bot_classifier.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ def test_classifies_bytespider(self):
8383
assert result['$is_ai_bot'] is True
8484
assert result['$ai_bot_name'] == 'Bytespider'
8585
assert result['$ai_bot_provider'] == 'ByteDance'
86+
assert result['$ai_bot_category'] == 'indexing'
8687

8788
# === Common Crawl ===
8889

@@ -91,6 +92,7 @@ def test_classifies_ccbot(self):
9192
assert result['$is_ai_bot'] is True
9293
assert result['$ai_bot_name'] == 'CCBot'
9394
assert result['$ai_bot_provider'] == 'Common Crawl'
95+
assert result['$ai_bot_category'] == 'indexing'
9496

9597
# === Apple ===
9698

@@ -102,6 +104,7 @@ def test_classifies_applebot_extended(self):
102104
assert result['$is_ai_bot'] is True
103105
assert result['$ai_bot_name'] == 'Applebot-Extended'
104106
assert result['$ai_bot_provider'] == 'Apple'
107+
assert result['$ai_bot_category'] == 'indexing'
105108

106109
# === Meta ===
107110

@@ -110,6 +113,7 @@ def test_classifies_meta_external_agent(self):
110113
assert result['$is_ai_bot'] is True
111114
assert result['$ai_bot_name'] == 'Meta-ExternalAgent'
112115
assert result['$ai_bot_provider'] == 'Meta'
116+
assert result['$ai_bot_category'] == 'indexing'
113117

114118
# === Cohere ===
115119

0 commit comments

Comments
 (0)