yams/tests/common/search_corpus_presets.h
2025-10-04 21:12:19 -04:00

70 lines
2.6 KiB
C++

#pragma once
#include <vector>
#include "fixture_manager.h"
namespace yams::test {
inline SearchCorpusSpec defaultSearchCorpusSpec() {
SearchCorpusSpec spec;
spec.rootDirectory = "corpus";
spec.bytesPerDocument = 384;
spec.commonTags = {"search", "corpus"};
spec.topics = {{.name = "quick",
.documentsPerTopic = 2,
.extension = ".txt",
.keywords = {"quick brown fox", "lazy dog"},
.tags = {"stories"}},
{.name = "python",
.documentsPerTopic = 2,
.extension = ".txt",
.keywords = {"Python programming language", "data science"},
.tags = {"code"}},
{.name = "machine",
.documentsPerTopic = 2,
.extension = ".txt",
.keywords = {"machine learning", "artificial intelligence"},
.tags = {"ml"}},
{.name = "deployment",
.documentsPerTopic = 1,
.extension = ".txt",
.keywords = {"quick deployment process"},
.tags = {"ops"}}};
return spec;
}
inline std::vector<std::string> defaultSearchQueries() {
return {"quick", "python", "machine", "deployment", "qwick",
"hello_world()", "{\"server\":", "30,New", "**bold**"};
}
inline SearchCorpusSpec mobileSearchCorpusSpec() {
SearchCorpusSpec spec;
spec.rootDirectory = "mobile";
spec.bytesPerDocument = 256;
spec.commonTags = {"mobile", "fixture"};
spec.topics = {{.name = "sync",
.documentsPerTopic = 2,
.extension = ".txt",
.keywords = {"offline sync", "delta"},
.tags = {"sync", "scenario:offline"}},
{.name = "case",
.documentsPerTopic = 2,
.extension = ".md",
.keywords = {"CaseSensitive", "casesensitive"},
.tags = {"case", "scenario:case"}},
{.name = "paths",
.documentsPerTopic = 2,
.extension = ".json",
.keywords = {"/Users/demo/Documents", "C:/Users/Demo/Documents"},
.tags = {"paths", "scenario:path"}},
{.name = "semantic",
.documentsPerTopic = 1,
.extension = ".txt",
.keywords = {"neural embedding warm up"},
.tags = {"semantic", "scenario:warmup"}}};
return spec;
}
} // namespace yams::test