1 files changed, 39 insertions, 0 deletions
diff --git a/src/test/ConnectedComponentsTest.java b/src/test/ConnectedComponentsTest.java
new file mode 100644
index 0000000..2e0888c
--- /dev/null
+++ b/src/test/ConnectedComponentsTest.java
@@ -0,0 +1,39 @@
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class ConnectedComponentsTest {
+
+    @Test
+    void test() {
+        ArrayList<ArrayList<Integer>> adj = new ArrayList<>();
+        for (int i = 0; i < 4; i++) {
+            adj.add(new ArrayList<>());
+        }
+
+        adj.get(0).add(1);
+        adj.get(2).add(1);
+        adj.get(3).add(2);
+        adj.get(0).add(3);
+
+        assertEquals(1, ConnectedComponents.numberOfComponents(adj));
+    }
+
+    @Test
+    void test1() {
+        ArrayList<ArrayList<Integer>> adj = new ArrayList<>();
+        for (int i = 0; i < 4; i++) {
+            adj.add(new ArrayList<>());
+        }
+
+        adj.get(0).add(1);
+        adj.get(1).add(0);
+        adj.get(2).add(1);
+        adj.get(1).add(2);
+
+        assertEquals(2, ConnectedComponents.numberOfComponents(adj));
+    }
+
+}
+\ No newline at end of file