Created TwoSum.py

TwoSum.py

@@ -0,0 +1,13 @@
+def two_sum(nums, target):
+    num_dict = {}
+    for i, num in enumerate(nums):
+        complement = target - num
+        if complement in num_dict:
+            return [num_dict[complement], i]
+        num_dict[num] = i
+    return None
+
+nums = [2, 7, 11, 15]
+target = 9
+result = two_sum(nums, target)
+print(result)  

twosum.py

@@ -0,0 +1,24 @@
+def two_sum(nums, target):
+    num_dict = {}
+    for i, num in enumerate(nums):
+        complement = target - num
+        if complement in num_dict:
+            return [num_dict[complement], i]
+        num_dict[num] = i
+    return None
+
+if __name__ == "__main__":
+    nums = [2, 7, 11, 15]
+    target = 9
+    result = two_sum(nums, target)
+    print(result) 
+
+def test_two_sum():
+    assert two_sum([2, 7, 11, 15], 9) == [0, 1]
+    assert two_sum([1, 2, 3, 4, 5], 9) == [3, 4]
+    assert two_sum([3, 2, 4], 6) == [1, 2]
+    assert two_sum([3, 3], 6) == [0, 1]
+    assert two_sum([], 5) is None
+    print("All tests passed.")
+
+test_two_sum()