IndexGraphBuilder Test: test edgeSource/edgeTarget, also for invalid edge ids

Author: barakugav

jgalgo-core/src/test/java/com/jgalgo/graph/IndexGraphBuilderTest.java

@@ -35,8 +35,11 @@
 import org.junit.jupiter.api.Test;
 import com.jgalgo.internal.util.TestBase;
 import it.unimi.dsi.fastutil.Pair;
+import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
+import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
 import it.unimi.dsi.fastutil.ints.IntArrayList;
 import it.unimi.dsi.fastutil.ints.IntArrays;
+import it.unimi.dsi.fastutil.ints.IntIntPair;
 import it.unimi.dsi.fastutil.ints.IntList;
 import it.unimi.dsi.fastutil.ints.IntSet;
 import it.unimi.dsi.fastutil.ints.IntSets;
@@ -495,6 +498,30 @@ public void addEdgesReassignIdsInvalidEndpoint() {
 		});
 	}
 
+	@Test
+	public void edgeSourceAndTarget() {
+		final Random rand = new Random(0x843e29cab46a15b0L);
+		foreachBoolConfig(directed -> {
+			IndexGraphBuilderImpl b = new IndexGraphBuilderImpl(directed);
+			b.addVertices(range(50));
+			Int2ObjectMap<IntIntPair> edges = new Int2ObjectOpenHashMap<>();
+			for (int i = 0; i < 50; i++) {
+				int u = rand.nextInt(50), v = rand.nextInt(50);
+				int e = b.addEdge(u, v);
+				edges.put(e, IntIntPair.of(u, v));
+			}
+			for (int e : b.edges()) {
+				IntIntPair endpoints = edges.get(e);
+				assertEquals(endpoints.firstInt(), b.edgeSource(e));
+				assertEquals(endpoints.secondInt(), b.edgeTarget(e));
+			}
+			for (int repeat = 0; repeat < 5; repeat++)
+				assertThrows(NoSuchEdgeException.class, () -> b.edgeSource(b.edges().size() + rand.nextInt(10)));
+			for (int repeat = 0; repeat < 5; repeat++)
+				assertThrows(NoSuchEdgeException.class, () -> b.edgeTarget(b.edges().size() + rand.nextInt(10)));
+		});
+	}
+
 	@Test
 	public void clear() {
 		foreachBoolConfig(directed -> {