lc: Add unittest to two sum

XuShaohua · Nov 14, 2023 · db8b151 · db8b151
1 parent d8f7a9a
commit db8b151
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/leetcode/001.two_sum/src/main.rs b/leetcode/001.two_sum/src/main.rs
@@ -28,3 +28,19 @@ fn main() {
     let target = 6;
     assert_eq!(solution1(nums, target), Some((0, 1)));
 }
+
+#[cfg(test)]
+mod tests {
+    use super::solution1;
+
+    #[test]
+    fn test_solution1() {
+        let nums = &[0, 4, 3, 0];
+        let target = 0;
+        assert_eq!(solution1(nums, target), Some((0, 3)));
+
+        let nums = &[-3, 4, 3, 90];
+        let target = 0;
+        assert_eq!(solution1(nums, target), Some((0, 2)));
+    }
+}