1fd9967558
This package provides a way to convert JSON schemas to equivalent EBNF. It is intended to be a replacement to llama.cpp's schema_to_grammar. This is still an early version and does not yet support all JSON schema features. The to-do list includes: - minumum/maximum constraints on integer types - minLength/maxLength constraints on string types - defs and refs
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
// Copyright 2022 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package diff
|
|
|
|
import (
|
|
"bytes"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"golang.org/x/tools/txtar"
|
|
)
|
|
|
|
func clean(text []byte) []byte {
|
|
text = bytes.ReplaceAll(text, []byte("$\n"), []byte("\n"))
|
|
text = bytes.TrimSuffix(text, []byte("^D\n"))
|
|
return text
|
|
}
|
|
|
|
func Test(t *testing.T) {
|
|
files, _ := filepath.Glob("testdata/*.txt")
|
|
if len(files) == 0 {
|
|
t.Fatalf("no testdata")
|
|
}
|
|
|
|
for _, file := range files {
|
|
t.Run(filepath.Base(file), func(t *testing.T) {
|
|
a, err := txtar.ParseFile(file)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(a.Files) != 3 || a.Files[2].Name != "diff" {
|
|
t.Fatalf("%s: want three files, third named \"diff\"", file)
|
|
}
|
|
diffs := Diff(a.Files[0].Name, clean(a.Files[0].Data), a.Files[1].Name, clean(a.Files[1].Data))
|
|
want := clean(a.Files[2].Data)
|
|
if !bytes.Equal(diffs, want) {
|
|
t.Fatalf("%s: have:\n%s\nwant:\n%s\n%s", file,
|
|
diffs, want, Diff("have", diffs, "want", want))
|
|
}
|
|
})
|
|
}
|
|
}
|