| from __future__ import annotations |
|
|
| from ctx.core.graph.edge_scoring import ( |
| SLUG_STOP, |
| adamic_adar_scores, |
| canonical_pair, |
| pairs_from_index, |
| slug_tokens, |
| type_affinity_score, |
| ) |
|
|
|
|
| def test_slug_tokens_filters_short_and_stop_words() -> None: |
| assert slug_tokens("fastapi-pro-skill") == ["fastapi"] |
| assert "pro" in SLUG_STOP |
|
|
|
|
| def test_pairs_from_index_skips_dense_buckets_and_tracks_shared_keys() -> None: |
| counts, shared = pairs_from_index( |
| { |
| "python": ["skill:b", "skill:a"], |
| "too-dense": ["skill:a", "skill:b", "skill:c"], |
| }, |
| dense_threshold=2, |
| ) |
|
|
| pair = ("skill:a", "skill:b") |
| assert counts[pair] == 1 |
| assert shared[pair] == ["python"] |
| assert ("skill:a", "skill:c") not in counts |
|
|
|
|
| def test_adamic_adar_scores_only_scores_candidate_pairs() -> None: |
| pair = canonical_pair("skill:a", "skill:b") |
| scores = adamic_adar_scores( |
| ["skill:a", "skill:b", "skill:c"], |
| {pair, canonical_pair("skill:a", "skill:c"), canonical_pair("skill:b", "skill:c")}, |
| ) |
|
|
| assert 0 < scores[pair] <= 1.0 |
|
|
|
|
| def test_type_affinity_score_matches_existing_graph_contract() -> None: |
| assert type_affinity_score("skill", "agent") == 1.0 |
| assert type_affinity_score("skill", "mcp-server") == 0.9 |
| assert type_affinity_score("skill", "skill") == 0.35 |
|
|
