determining_dna_health_test.go

// Package determining_dna_health contains the solution for HackerRank problem
// https://www.hackerrank.com/challenges/determining-dna-health/problem.
package determining_dna_health_test

import (
	"bufio"
	"io/ioutil"
	"os"
	"strconv"
	"strings"
	"testing"

	"github.com/sepetrov/hackerrank/determining_dna_health"
)

const bufSize = 1024 * 1024

func TestRun(t *testing.T) {
	type out struct {
		min, max int
	}
	parseOutput := func(filename string) out {
		body, err := ioutil.ReadFile(filename)
		if err != nil {
			panic(err)
		}
		mm := strings.Fields(string(body))
		min, err := strconv.Atoi(mm[0])
		if err != nil {
			panic(err)
		}
		max, err := strconv.Atoi(mm[1])
		if err != nil {
			panic(err)
		}
		return out{min, max}
	}
	tests := []struct {
		in  string
		out out
	}{
		{
			"testdata/input.txt",
			parseOutput("testdata/output.txt"),
		},
		// {
		// 	"testdata/input02.txt",
		// 	parseOutput("testdata/output02.txt"),
		// },
		// {
		// 	"testdata/input07.txt",
		// 	parseOutput("testdata/output07.txt"),
		// },
	}
	for _, tt := range tests {
		t.Run(tt.in, func(t *testing.T) {
			f, err := os.Open(tt.in)
			if err != nil {
				t.Fatal(err)
			}
			defer f.Close()
			min, max := determining_dna_health.Run(bufio.NewReaderSize(f, bufSize))

			if min != tt.out.min {
				t.Errorf("Run() min = %d, want %d", min, tt.out.min)
			}
			if max != tt.out.max {
				t.Errorf("Run() max = %d, want %d", max, tt.out.max)
			}
		})
	}
}

func BenchmarkRun(b *testing.B) {
	for _, f := range []string{
		"testdata/input.txt",
		"testdata/input02.txt",
		"testdata/input07.txt",
	} {
		b.Run(f, func(b *testing.B) {
			f, err := os.Open(f)
			if err != nil {
				b.Fatal(err)
			}
			defer f.Close() // In case Run() panics before we can close f.
			buf := bufio.NewReaderSize(f, bufSize)
			b.ResetTimer()
			for n := 0; n < b.N; n++ {
				determining_dna_health.Run(buf)
			}
			f.Close()
		})
	}
}