fix race condition in windows_test.go

cmdoret · Oct 7, 2020 · 13430ca · 13430ca
1 parent f8f4a57
commit 13430ca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dnaglider/pkg/windows_test.go b/dnaglider/pkg/windows_test.go
@@ -43,13 +43,13 @@ func TestChunkGenome(t *testing.T) {
 	}
 }
 func TestConsumeChunks(t *testing.T) {
-	records := StreamGenome(TESTFILE, 1)
 	ref := make(map[int]KmerProfile)
 	ref[3] = FastaToKmers(TESTFILE, 3)
+	records := StreamGenome(TESTFILE, 1)
 	chunks := ChunkGenome(records, WINSIZE, WINSTRIDE, CHUNKSIZE)
 	expHeader := []string{"chrom", "start", "end", "GC", "GCSKEW", "ENTRO", "3MER"}
-	results := ConsumeChunks(chunks, []string{"GC", "GCSKEW", "ENTRO"}, ref)
-	for res := range results {
+
+	for res := range ConsumeChunks(chunks, []string{"GC", "GCSKEW", "ENTRO"}, ref) {
 		for i := range res.Header {
 			if res.Header[i] != expHeader[i] {
 				t.Errorf("Result header is incorrect.: %s", res.Header)