2023/go/day02/day02_test.go

   1 package day02
   2
   3 import (
   4         "testing"
   5
   6         "github.com/stretchr/testify/require"
   7 )
   8
   9 func TestPart1(t *testing.T) {
  10         r := Part1(
  11 `Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green
  12 Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue
  13 Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red
  14 Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red
  15 Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green`)
  16         require.Equal(t, 8, r)
  17 }
  18
  19 func TestPart2(t *testing.T) {
  20         r := Part2(
  21                 `Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green
  22                 Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue
  23                 Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red
  24                 Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red
  25                 Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green`)
  26         require.Equal(t, 2286, r)
  27 }