diff --git a/src/main/java/com/thealgorithms/maths/Combinations.java b/src/main/java/com/thealgorithms/maths/Combinations.java index eee9b9fc8..34b324466 100644 --- a/src/main/java/com/thealgorithms/maths/Combinations.java +++ b/src/main/java/com/thealgorithms/maths/Combinations.java @@ -5,23 +5,6 @@ package com.thealgorithms.maths; */ public class Combinations { - public static void main(String[] args) { - assert combinations(1, 1) == 1; - assert combinations(10, 5) == 252; - assert combinations(6, 3) == 20; - assert combinations(20, 5) == 15504; - - // Since, 200 is a big number its factorial will go beyond limits of long even when 200C5 can be saved in a long - // variable. So below will fail - // assert combinations(200, 5) == 2535650040l; - assert combinationsOptimized(100, 0) == 1; - assert combinationsOptimized(1, 1) == 1; - assert combinationsOptimized(10, 5) == 252; - assert combinationsOptimized(6, 3) == 20; - assert combinationsOptimized(20, 5) == 15504; - assert combinationsOptimized(200, 5) == 2535650040l; - } - /** * Calculate of factorial * diff --git a/src/test/java/com/thealgorithms/maths/CombinationsTest.java b/src/test/java/com/thealgorithms/maths/CombinationsTest.java new file mode 100644 index 000000000..0bcefb7de --- /dev/null +++ b/src/test/java/com/thealgorithms/maths/CombinationsTest.java @@ -0,0 +1,30 @@ +package com.thealgorithms.maths; + +import org.junit.jupiter.api.Test; + +import static org.junit.jupiter.api.Assertions.assertEquals; + +public class CombinationsTest { + + @Test + void testCombination() { + + assertEquals(1, Combinations.combinations(1, 1)); + assertEquals(252, Combinations.combinations(10, 5)); + assertEquals(20, Combinations.combinations(6, 3)); + assertEquals(15504, Combinations.combinations(20, 5)); + + } + + @Test + void testCombinationOptimised() { + + assertEquals(100, Combinations.combinationsOptimized(100, 1)); + assertEquals(1, Combinations.combinationsOptimized(1, 1)); + assertEquals(252, Combinations.combinationsOptimized(10, 5)); + assertEquals(20, Combinations.combinationsOptimized(6, 3)); + assertEquals(15504, Combinations.combinationsOptimized(20, 5)); + assertEquals(2535650040L, Combinations.combinationsOptimized(200, 5)); + + } +}