From 3086cc00d5b1d366ff74eb385dd28c0d242c4cc9 Mon Sep 17 00:00:00 2001 From: Thomas Fabre <tfabre@takima.fr> Date: Fri, 23 Jun 2023 14:29:51 +0200 Subject: [PATCH] Refactor largeStraight with Roll --- src/main/java/io/takima/Yatzy.java | 17 ++++------------- src/test/java/io/takima/YatzyTest.java | 8 ++++---- 2 files changed, 8 insertions(+), 17 deletions(-) diff --git a/src/main/java/io/takima/Yatzy.java b/src/main/java/io/takima/Yatzy.java index c75cead..6150c2f 100644 --- a/src/main/java/io/takima/Yatzy.java +++ b/src/main/java/io/takima/Yatzy.java @@ -6,6 +6,7 @@ import java.util.stream.IntStream; public final class Yatzy { private static final List<Integer> SMALL_STRAIGHT = IntStream.range(1, 6).boxed().toList(); + private static final List<Integer> LARGE_STRAIGHT = IntStream.range(2, 7).boxed().toList(); private Yatzy() { // This his a utility class which only implements static methods @@ -94,19 +95,9 @@ public final class Yatzy { return 0; } - public static int largeStraight(int d1, int d2, int d3, int d4, int d5) { - int[] counts; - counts = new int[6]; - counts[d1 - 1] += 1; - counts[d2 - 1] += 1; - counts[d3 - 1] += 1; - counts[d4 - 1] += 1; - counts[d5 - 1] += 1; - if (counts[1] == 1 && - counts[2] == 1 && - counts[3] == 1 && - counts[4] == 1 - && counts[5] == 1) { + public static int largeStraight(Roll roll) { + List<Integer> sortedRoll = roll.toStream().sorted().boxed().toList(); + if (sortedRoll.equals(LARGE_STRAIGHT)) { return 20; } return 0; diff --git a/src/test/java/io/takima/YatzyTest.java b/src/test/java/io/takima/YatzyTest.java index 2a741e1..45b0ff7 100644 --- a/src/test/java/io/takima/YatzyTest.java +++ b/src/test/java/io/takima/YatzyTest.java @@ -151,16 +151,16 @@ class YatzyTest { @Test void largeStraightShouldReturnTwentyIfDicesAreFromTwoToSix() { assertAll( - () -> assertEquals(20, Yatzy.largeStraight(6, 2, 3, 4, 5)), - () -> assertEquals(20, Yatzy.largeStraight(2, 3, 4, 5, 6)) + () -> assertEquals(20, Yatzy.largeStraight(roll(6, 2, 3, 4, 5))), + () -> assertEquals(20, Yatzy.largeStraight(roll(2, 3, 4, 5, 6))) ); } @Test void largeStraightShouldReturnZeroIfDicesAreNotFromTwoToSix() { assertAll( - () -> assertEquals(0, Yatzy.largeStraight(1, 2, 2, 4, 5)), - () -> assertEquals(0, Yatzy.largeStraight(2, 4, 6, 5, 1)) + () -> assertEquals(0, Yatzy.largeStraight(roll(1, 2, 2, 4, 5))), + () -> assertEquals(0, Yatzy.largeStraight(roll(2, 4, 6, 5, 1))) ); } -- GitLab