Cover test case #2

Author: balazs-domonkos-epam

src/string_calculator.py

@@ -1,4 +1,7 @@
 def calculate_sum(input_string):
-    numbers = input_string.split(',')
-    number1, number2 = map(int, numbers)
-    return number1 + number2
+    if input_string is None or input_string == "":
+        return 0
+    else:
+        numbers = input_string.split(',')
+        number1, number2 = map(int, numbers)
+        return number1 + number2

test/test_string_calculator.py

@@ -16,7 +16,17 @@
     ("5,99", 104)
 ]
 
+test_cases_for_empty_or_null_input = [
+    ("", 0),
+    (None, 0)
+]
+
 
 @pytest.mark.parametrize("input_string, expected", test_cases_for_sum_calculation)
 def test_sum_calculation(input_string, expected):
     assert calculate_sum(input_string) == expected
+
+
+@pytest.mark.parametrize("input_string, expected", test_cases_for_empty_or_null_input)
+def test_empty_or_null_input(input_string, expected):
+    assert calculate_sum(input_string) == expected