Commit d08f030

committed

Test 2024 day 22 examples with scalacheck

1 parent 0103bd6 commit d08f030Copy full SHA for d08f030

File tree

+27

-16

lines changed

src
- main/scala/eu/sim642/adventofcode2024
  - Day22.scala
- test/scala/eu/sim642/adventofcode2024
  - Day22Test.scala

+27

-16

lines changed

Lines changed: 4 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -15,17 +15,17 @@ object Day22 {`
`15`	`15`	`prune(mix(secret2, secret2 * 2048))`
`16`	`16`	`}`
`17`	`17`
`18`		`- def secretAfter(initialSecret: Secret, after: Int=2000): Secret =`
`19`		`- Iterator.iterate(initialSecret)(nextSecret)(after)`
	`18`	`+ def secretIterator(initialSecret: Secret): Iterator[Secret] =`
	`19`	`+ Iterator.iterate(initialSecret)(nextSecret)`
`20`	`20`
`21`	`21`	`def sumSecretsAfter(secrets: Seq[Secret], after: Int = 2000): Secret =`
`22`		`- secrets.map(secretAfter(_, after)).sum`
	`22`	`+ secrets.map(secretIterator(_)(after)).sum`
`23`	`23`
`24`	`24`	`def mostBananas(secrets: Seq[Secret]): Int = {`
`25`	`25`	`// TODO: optimize (~4.7s)`
`26`	`26`	`val secretMaps = secrets`
`27`	`27`	`.map({ initialSecret =>`
`28`		`- Iterator.iterate(initialSecret, 2000 + 1)(nextSecret)`
	`28`	`+ secretIterator(initialSecret).take(2000 + 1)`
`29`	`29`	`.map(_ % 10)`
`30`	`30`	`.map(_.toInt)`
`31`	`31`	`.sliding(5)`

Lines changed: 23 additions & 12 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,9 +1,10 @@`
`1`	`1`	`package eu.sim642.adventofcode2024`
`2`	`2`
`3`		`-import Day22._`
	`3`	`+import Day22.*`
`4`	`4`	`import org.scalatest.funsuite.AnyFunSuite`
	`5`	`+import org.scalatestplus.scalacheck.ScalaCheckPropertyChecks`
`5`	`6`
`6`		`-class Day22Test extends AnyFunSuite {`
	`7`	`+class Day22Test extends AnyFunSuite withScalaCheckPropertyChecks{`
`7`	`8`
`8`	`9`	`val exampleInput =`
`9`	`10`	`"""1`
`@@ -18,16 +19,26 @@ class Day22Test extends AnyFunSuite {`
`18`	`19`	`\|2024""".stripMargin`
`19`	`20`
`20`	`21`	`test("Part 1 examples") {`
`21`		`- assert(secretAfter(123, 1) == 15887950)`
`22`		`- assert(secretAfter(123, 2) == 16495136)`
`23`		`- assert(secretAfter(123, 3) == 527345)`
`24`		`- assert(secretAfter(123, 4) == 704524)`
`25`		`- assert(secretAfter(123, 5) == 1553684)`
`26`		`- assert(secretAfter(123, 6) == 12683156)`
`27`		`- assert(secretAfter(123, 7) == 11100544)`
`28`		`- assert(secretAfter(123, 8) == 12249484)`
`29`		`- assert(secretAfter(123, 9) == 7753432)`
`30`		`- assert(secretAfter(123, 10) == 5908254)`
	`22`	`+ val secrets = Table(`
	`23`	`+ "secret",`
	`24`	`+ 123,`
	`25`	`+ 15887950,`
	`26`	`+ 16495136,`
	`27`	`+ 527345,`
	`28`	`+ 704524,`
	`29`	`+ 1553684,`
	`30`	`+ 12683156,`
	`31`	`+ 11100544,`
	`32`	`+ 12249484,`
	`33`	`+ 7753432,`
	`34`	`+ 5908254,`
	`35`	`+ )`
	`36`	`+`
	`37`	`+ val it = secretIterator(123)`
	`38`	`+ forAll (secrets) { secret =>`
	`39`	`+ assert(it.next() == secret)`
	`40`	`+ }`
	`41`	`+`
`31`	`42`	`assert(sumSecretsAfter(parseSecrets(exampleInput)) == 37327623)`
`32`	`43`	`}`
`33`	`44`

Comments

(0)