Add MultiAgentScheduling algorithm (#5816)

This commit is contained in:
Hardik Pawar
2024-10-26 12:23:36 +05:30
committed by GitHub
parent 92a04f8f8d
commit 32cb98db01
3 changed files with 126 additions and 0 deletions

View File

@@ -0,0 +1,52 @@
package com.thealgorithms.scheduling;
import static org.junit.jupiter.api.Assertions.assertEquals;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
public class MultiAgentSchedulingTest {
private MultiAgentScheduling scheduler;
@BeforeEach
public void setup() {
scheduler = new MultiAgentScheduling();
}
@Test
public void testAddAgentAndAssignTask() {
scheduler.addAgent("Agent1");
scheduler.assignTask("Agent1", "Task1");
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1"));
assertEquals(expected, scheduler.getScheduledTasks());
}
@Test
public void testMultipleAgentsWithTasks() {
scheduler.addAgent("Agent1");
scheduler.addAgent("Agent2");
scheduler.assignTask("Agent1", "Task1");
scheduler.assignTask("Agent2", "Task2");
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1"), "Agent2", List.of("Task2"));
assertEquals(expected, scheduler.getScheduledTasks());
}
@Test
public void testAgentWithMultipleTasks() {
scheduler.addAgent("Agent1");
scheduler.assignTask("Agent1", "Task1");
scheduler.assignTask("Agent1", "Task2");
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1", "Task2"));
assertEquals(expected, scheduler.getScheduledTasks());
}
@Test
public void testEmptyAgentSchedule() {
scheduler.addAgent("Agent1");
Map<String, List<String>> expected = Map.of("Agent1", List.of());
assertEquals(expected, scheduler.getScheduledTasks());
}
}