wehub-resource-sync
2860fb5d18
Security / Dependency review (push) Has been skipped
Scorecard / Scorecard analysis (push) Failing after 0s
Validate / eval (push) Failing after 0s
Security / Dependency audit (push) Failing after 1s
Security / Secret scan (push) Failing after 1s
Validate / tests (push) Failing after 0s
Validate / mcp-tests (push) Failing after 1s
GitHub Actions Security Analysis with zizmor 🌈 / zizmor (push) Failing after 1s
Security / SAST scan (push) Failing after 13m52s
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import unittest
|
|
|
|
from lib import entity_extract
|
|
|
|
|
|
class TestExtractSubreddits(unittest.TestCase):
|
|
def test_extracts_primary_subreddit(self):
|
|
items = [{"subreddit": "r/MachineLearning"}]
|
|
result = entity_extract._extract_subreddits(items)
|
|
self.assertIn("MachineLearning", result)
|
|
|
|
def test_extracts_subreddit_without_prefix(self):
|
|
items = [{"subreddit": "localLLaMA"}]
|
|
result = entity_extract._extract_subreddits(items)
|
|
self.assertIn("localLLaMA", result)
|
|
|
|
def test_extracts_cross_references_from_comment_insights(self):
|
|
items = [
|
|
{
|
|
"subreddit": "technology",
|
|
"comment_insights": ["check out r/MachineLearning and r/LocalLLaMA for more"],
|
|
}
|
|
]
|
|
result = entity_extract._extract_subreddits(items)
|
|
self.assertIn("MachineLearning", result)
|
|
self.assertIn("LocalLLaMA", result)
|
|
|
|
def test_extracts_cross_references_from_top_comments(self):
|
|
items = [
|
|
{
|
|
"subreddit": "AI",
|
|
"top_comments": [
|
|
{"excerpt": "see r/StableDiffusion for image gen stuff"},
|
|
],
|
|
}
|
|
]
|
|
result = entity_extract._extract_subreddits(items)
|
|
self.assertIn("StableDiffusion", result)
|
|
|
|
def test_ranks_by_frequency(self):
|
|
items = [
|
|
{"subreddit": "A"},
|
|
{"subreddit": "A"},
|
|
{"subreddit": "B"},
|
|
]
|
|
result = entity_extract._extract_subreddits(items)
|
|
self.assertEqual(result[0], "A")
|
|
|
|
def test_empty_items(self):
|
|
self.assertEqual([], entity_extract._extract_subreddits([]))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|