scripts/bench_comp: fixup benchmark script

Author: charlievieth

scripts/bench_comp.go

@@ -4,7 +4,6 @@ import (
 	"flag"
 	"fmt"
 	"io"
-	"io/ioutil"
 	"log"
 	"os"
 	"os/exec"
@@ -17,29 +16,24 @@ func init() {
 	log.SetFlags(log.LstdFlags | log.Lshortfile)
 }
 
-var Tests = []string{
-	"filepath",
-	"fastwalk",
-}
-
 func main() {
-	count := flag.Int("count", 5, "Run each test and benchmark n times")
+	count := flag.Int("count", 10, "Run each test and benchmark n times")
 	compCmd := flag.String("comp", "benchstat", "Benchmark comparison command")
 	flag.Parse()
 
 	if _, err := exec.LookPath(*compCmd); err != nil {
 		log.Fatalf("error: %v: %q\n", err, *compCmd)
 	}
 
-	tmpdir, err := ioutil.TempDir("", "fastwalk-bench.*")
+	tmpdir, err := os.MkdirTemp("", "fastwalk-bench.*")
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	runTest := func(name string) error {
 		fmt.Println("##", name)
 
-		filename := filepath.Join(tmpdir, name+".txt")
+		filename := filepath.Join(tmpdir, name)
 		f, err := os.Create(filename)
 		if err != nil {
 			log.Fatal(err)
@@ -52,7 +46,6 @@ func main() {
 			"-bench", `^BenchmarkWalkComparison$`,
 			"-benchmem",
 			"-count", strconv.Itoa(*count),
-			"github.com/charlievieth/fastwalk",
 			"-walkfunc", name,
 		}
 
@@ -72,18 +65,15 @@ func main() {
 		return nil
 	}
 
-	for _, name := range Tests {
-		runTest(name)
-	}
+	runTest("filepath")
+	runTest("fastwalk")
 
 	benchStat := func(from, to string) {
 		fmt.Printf("## %s vs. %s\n", from, to)
-		cmd := exec.Command(*compCmd,
-			filepath.Join(tmpdir, from+".txt"),
-			filepath.Join(tmpdir, to+".txt"),
-		)
+		cmd := exec.Command(*compCmd, from, to)
 		cmd.Stderr = os.Stderr
 		cmd.Stdout = os.Stdout
+		cmd.Dir = tmpdir
 		if err := cmd.Run(); err != nil {
 			log.Fatalf("error running command: %q: %v\n", cmd.Args, err)
 		}
@@ -94,7 +84,6 @@ func main() {
 	fmt.Println("########################################################")
 	fmt.Print("\n")
 	benchStat("filepath", "fastwalk")
-	benchStat("godirwalk", "fastwalk")
 
 	fmt.Printf("Temp: %s\n", tmpdir)
 }