32 lines
1.3 KiB
Go
32 lines
1.3 KiB
Go
package server
|
|
|
|
import "testing"
|
|
|
|
func TestIsSoftwareDevelopment(t *testing.T) {
|
|
tests := []struct {
|
|
message string
|
|
expected bool
|
|
}{
|
|
{"can you check the logs? it looks like a lot of requests are being routed to kimi-k2.7-code when they don't need to be", false},
|
|
{"it looks like its still routing to kimi when it shouldn't need to", false},
|
|
{"Write a python script to parse logs", true},
|
|
{"Search the web for weather in New York", false},
|
|
{"How to build a compiler in Go", true},
|
|
{"Check my vscode config", false},
|
|
{"Let's decode this barcode", false},
|
|
{"go ahead and email both docxs and the ppt to kayla\n\n<memory-context>... HELPipedia_Telethon_Speaker_Script.docx ... mental health program specialist ...", false},
|
|
{"academic program development", false},
|
|
{"write a script for the video presentation", false},
|
|
{"How to write a bash script", true},
|
|
{"refactor this sql query", true},
|
|
{"can you organize Downloads and Documents real quick?\n\n<memory-context>\nCLI wrapper: ~/Projects/rag-engine/rag (bash script activating .venv). Python venv@~/Projects/rag-engine/.venv/.\n</memory-context>", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
result := isSoftwareDevelopment(tt.message)
|
|
if result != tt.expected {
|
|
t.Errorf("isSoftwareDevelopment(%q) = %v; expected %v", tt.message, result, tt.expected)
|
|
}
|
|
}
|
|
}
|