diff --git a/src/stringcalculator.py b/src/stringcalculator.py index a1463b3c7c433283b37efb5099204564f965fd7b..4b8e76f33e53fb4ef8cdc900d5930821fb39baed 100644 --- a/src/stringcalculator.py +++ b/src/stringcalculator.py @@ -27,4 +27,6 @@ class StringCalculator(IStringCalculator): if negative_numbers: raise ValueError(f"Negative nicht erlaubt: {negative_numbers}") + numbers_list = [num for num in numbers_list if num <= 1000] + return sum(numbers_list) \ No newline at end of file diff --git a/tests/test_stringclaculator.py b/tests/test_stringclaculator.py index d72bf6e90b5ea741a43520258665a7a5df6c2cb0..a0c1d7668215796feb20d932160f840f2e5a09db 100644 --- a/tests/test_stringclaculator.py +++ b/tests/test_stringclaculator.py @@ -67,6 +67,10 @@ class TestStringCalculator(unittest.TestCase): self.calculator.add("//;1;2") print(str(context.exception)) + def test_ignore_numbers_greater_than_1000(self): + self.assertEqual(self.calculator.add("2,1001"), 2) + self.assertEqual(self.calculator.add("1002,50200"), 0) + if __name__ == "__main__": unittest.main()