42 lines
619 B
Go
42 lines
619 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
_ "embed"
|
||
|
|
|
||
|
|
"github.com/matryer/is"
|
||
|
|
)
|
||
|
|
|
||
|
|
//go:embed example.txt
|
||
|
|
var example []byte
|
||
|
|
|
||
|
|
//go:embed input.txt
|
||
|
|
var input []byte
|
||
|
|
|
||
|
|
func TestExample(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
scan := bufio.NewScanner(bytes.NewReader(example))
|
||
|
|
|
||
|
|
result, err := run(scan)
|
||
|
|
is.NoErr(err)
|
||
|
|
|
||
|
|
t.Log(result)
|
||
|
|
is.Equal(result.sum, 8)
|
||
|
|
is.Equal(result.powerSum, 2286)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSolution(t *testing.T) {
|
||
|
|
is := is.New(t)
|
||
|
|
scan := bufio.NewScanner(bytes.NewReader(input))
|
||
|
|
|
||
|
|
result, err := run(scan)
|
||
|
|
is.NoErr(err)
|
||
|
|
|
||
|
|
t.Log(result)
|
||
|
|
is.Equal(result.sum, 2317)
|
||
|
|
is.Equal(result.powerSum, 74804)
|
||
|
|
}
|