github.com/koraygocmen/MedianFilter@v0.0.0-20190901023803-fc0c97ab22ce/bench_test.go (about)

     1  package MedianFilter
     2  
     3  import (
     4  	"fmt"
     5  	"io/ioutil"
     6  	"log"
     7  	"os"
     8  	"testing"
     9  )
    10  
    11  func Benchmark(b *testing.B) {
    12  
    13  	testFolder := "test-frames/"
    14  	outPath := testFolder + "out.png"
    15  
    16  	var filePaths []string
    17  	files, err := ioutil.ReadDir(testFolder)
    18  	if err != nil {
    19  		log.Fatal(err)
    20  	}
    21  
    22  	for _, f := range files {
    23  		filePaths = append(filePaths, testFolder+f.Name())
    24  	}
    25  
    26  	if err := RemoveMovingObjs(filePaths, outPath); err != nil {
    27  		fmt.Println(err)
    28  	}
    29  
    30  	os.Remove(outPath)
    31  }