feat: initial commit

2026-01-01 03:40:41 +08:00
commit 631f8ed771
98 changed files with 14776 additions and 0 deletions
@@ -0,0 +1,91 @@
+using MaigoLabs.NeedLe.Indexer;
+using MaigoLabs.NeedLe.Searcher;
+
+namespace MaigoLabs.NeedLe.Tests.E2E;
+
+public sealed class Search_MatchesWithMixedSearchQueryTest : NeedleTestBase
+{
+    private static readonly string[] TestDocuments =
+    [
+        "ミーティア",
+        "エンドマークに希望と涙を添えて",
+        "宵の鳥",
+        "僕の和風本当上手",
+    ];
+
+    [Fact]
+    public void Execute()
+    {
+        var compressed = InvertedIndexBuilder.BuildInvertedIndex(TestDocuments, TokenizerOptions);
+        var invertedIndex = InvertedIndexLoader.Load(compressed);
+
+        var results = InvertedIndexSearcher.Search(invertedIndex, "bokunoh风じょう");
+
+        // Should have at least one result
+        Assert.NotEmpty(results);
+
+        // The first result should be "僕の和風本当上手"
+        Assert.Equal("僕の和風本当上手", results[0].DocumentText);
+    }
+}
+
+public sealed class Search_HighlightsSearchResultCorrectlyTest : NeedleTestBase
+{
+    private static readonly string[] TestDocuments =
+    [
+        "ミーティア",
+        "エンドマークに希望と涙を添えて",
+        "宵の鳥",
+        "僕の和風本当上手",
+    ];
+
+    [Fact]
+    public void Execute()
+    {
+        var compressed = InvertedIndexBuilder.BuildInvertedIndex(TestDocuments, TokenizerOptions);
+        var invertedIndex = InvertedIndexLoader.Load(compressed);
+
+        var results = InvertedIndexSearcher.Search(invertedIndex, "bokunoh风じょう");
+        Assert.NotEmpty(results);
+
+        var highlighted = SearchResultHighlighter.Highlight(results[0]);
+
+        // Should be a list of parts
+        Assert.NotEmpty(highlighted);
+
+        // Collect highlighted text
+        var highlightedTexts = highlighted.Where(p => p.IsHighlighted).Select(p => p.Text).ToList();
+        var highlightedJoined = string.Join("", highlightedTexts);
+
+        Assert.Contains("僕", highlightedJoined);
+        Assert.Contains("の", highlightedJoined);
+        Assert.Contains("和", highlightedJoined);
+        Assert.Contains("風", highlightedJoined);
+        Assert.Contains("上", highlightedJoined);
+    }
+}
+
+public sealed class Search_MatchesRomajiInputToKanaDocumentsTest : NeedleTestBase
+{
+    private static readonly string[] TestDocuments =
+    [
+        "ミーティア",
+        "エンドマークに希望と涙を添えて",
+        "宵の鳥",
+        "僕の和風本当上手",
+    ];
+
+    [Fact]
+    public void Execute()
+    {
+        var compressed = InvertedIndexBuilder.BuildInvertedIndex(TestDocuments, TokenizerOptions);
+        var invertedIndex = InvertedIndexLoader.Load(compressed);
+
+        // Search for "yoi" should match "宵の鳥"
+        var results = InvertedIndexSearcher.Search(invertedIndex, "yoi");
+        var matchedTexts = results.Select(r => r.DocumentText).ToList();
+
+        Assert.Contains("宵の鳥", matchedTexts);
+    }
+}
+