AssignmentUsingBitmaskTest.java

packagecom.thealgorithms.dynamicprogramming;

importstaticjava.util.Collections.singletonList;
importstaticorg.junit.jupiter.api.Assertions.assertEquals;

importjava.util.Arrays;
importjava.util.List;
importorg.junit.jupiter.api.Test;

publicfinalclassAssignmentUsingBitmaskTest {

@Test
publicvoidtestCountNoOfWays() {
inttotalTasks = 5;

List<List<Integer>> taskPerformed = Arrays.asList(Arrays.asList(1, 3, 4), Arrays.asList(1, 2, 5), Arrays.asList(3, 4));

AssignmentUsingBitmaskassignment = newAssignmentUsingBitmask(taskPerformed, totalTasks);
intways = assignment.countNoOfWays();
assertEquals(10, ways);
 }

@Test
publicvoidtestNoPossibleAssignments() {
inttotalTasks = 3;

List<List<Integer>> taskPerformed = Arrays.asList(singletonList(2), singletonList(3));

AssignmentUsingBitmaskassignment = newAssignmentUsingBitmask(taskPerformed, totalTasks);
intways = assignment.countNoOfWays();
assertEquals(1, ways);
 }

@Test
publicvoidtestSinglePersonMultipleTasks() {
inttotalTasks = 3;

List<List<Integer>> taskPerformed = singletonList(Arrays.asList(1, 2, 3));

AssignmentUsingBitmaskassignment = newAssignmentUsingBitmask(taskPerformed, totalTasks);
intways = assignment.countNoOfWays();
assertEquals(3, ways);
 }

@Test
publicvoidtestMultiplePeopleSingleTask() {
inttotalTasks = 1;

List<List<Integer>> taskPerformed = Arrays.asList(singletonList(1), singletonList(1));

AssignmentUsingBitmaskassignment = newAssignmentUsingBitmask(taskPerformed, totalTasks);
intways = assignment.countNoOfWays();
assertEquals(0, ways);
 }
}