github.com/hscells/guru@v0.0.0-20200207042420-2dabeb950d69/xtrecresults_test.go (about)

     1  package guru_test
     2  
     3  import (
     4  	"github.com/hscells/guru"
     5  	"github.com/hscells/trecresults"
     6  	"os"
     7  	"testing"
     8  )
     9  
    10  func TestWriteCompressedTrecResultFile(t *testing.T) {
    11  	fR, err := os.OpenFile("testdata/tar18_training.res", os.O_RDONLY, 0664)
    12  	if err != nil {
    13  		t.Fatal(err)
    14  	}
    15  	resultFile, err := trecresults.ResultsFromReader(fR)
    16  	if err != nil {
    17  		t.Fatal(err)
    18  	}
    19  	fC, err := os.OpenFile("testdata/tar18_training.xres", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0664)
    20  	if err != nil {
    21  		t.Fatal(err)
    22  	}
    23  	_, err = guru.WriteCompressedTrecResultFile(fC, resultFile)
    24  	if err != nil {
    25  		t.Fatal(err)
    26  	}
    27  }
    28  
    29  func TestReadCompressedTrecResultFile(t *testing.T) {
    30  	f, err := os.OpenFile("testdata/tar18_training.xres", os.O_RDONLY, 0664)
    31  	if err != nil {
    32  		t.Fatal(err)
    33  	}
    34  	resultFile, err := guru.ReadCompressedTrecResultFile(f)
    35  	if err != nil {
    36  		t.Fatal(err)
    37  	}
    38  	t.Log(len(resultFile.Results))
    39  	for k, v := range resultFile.Results {
    40  		t.Log(k, len(v))
    41  	}
    42  }
    43  
    44  func TestReadTrecResultFile(t *testing.T) {
    45  	f, err := os.OpenFile("testdata/tar18_training.res", os.O_RDONLY, 0664)
    46  	if err != nil {
    47  		t.Fatal(err)
    48  	}
    49  	resultFile, err := trecresults.ResultsFromReader(f)
    50  	if err != nil {
    51  		t.Fatal(err)
    52  	}
    53  	t.Log(len(resultFile.Results))
    54  	for k, v := range resultFile.Results {
    55  		t.Log(k, len(v))
    56  	}
    57  }