fastjet: embed test files to speedup race tests

Signed-off-by: Sebastien Binet <binet@cern.ch>
go-hep · Feb 18, 2025 · 2b7bcba · 2b7bcba
1 parent 34c208f
commit 2b7bcba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/fastjet/algorithm_test.go b/fastjet/algorithm_test.go
@@ -8,7 +8,6 @@ import (
 	"bufio"
 	"fmt"
 	"math"
-	"os"
 	"sort"
 	"strconv"
 	"strings"
@@ -365,7 +364,7 @@ func TestExclusiveJetAlgorithms(t *testing.T) {
 }
 
 func loadParticles(name string) ([]fastjet.Jet, error) {
-	f, err := os.Open(name)
+	f, err := testFS.Open(name)
 	if err != nil {
 		return nil, err
 	}
@@ -408,7 +407,7 @@ func loadParticles(name string) ([]fastjet.Jet, error) {
 }
 
 func loadRef(name string) ([][3]float64, error) {
-	f, err := os.Open(name)
+	f, err := testFS.Open(name)
 	if err != nil {
 		return nil, err
 	}

diff --git a/fastjet/fastjet_test.go b/fastjet/fastjet_test.go
@@ -5,13 +5,17 @@
 package fastjet_test
 
 import (
+	"embed"
 	"sort"
 	"testing"
 
 	"go-hep.org/x/hep/fastjet"
 	"go-hep.org/x/hep/fmom"
 )
 
+//go:embed testdata/*.ref testdata/*.dat
+var testFS embed.FS
+
 func TestSimple(t *testing.T) {
 	t.Parallel()