Add files via upload

Author: Nidhal-labri

arithmetic_arranger.py

@@ -0,0 +1,36 @@
+# This code was made by NIDHAL LABRI #
+def arithmetic_arranger(problems, show_answers=False):
+    if len(problems) > 5:
+        return "Error: Too many problems."
+
+    arranged_problems = ""
+    line1 = line2 = line3 = line4 = ""
+
+    for problem in problems:
+        num1, operator, num2 = problem.split()
+        #Errors check
+        if operator not in ['+', '-']:
+            return "Error: Operator must be '+' or '-'."
+        if not num1.isdigit() or not num2.isdigit():
+            return "Error: Numbers must only contain digits."
+        if len(num1) > 4 or len(num2) > 4:
+            return "Error: Numbers cannot be more than four digits."
+
+        width = max(len(num1), len(num2)) + 2
+        line1 += num1.rjust(width) + "    "
+        line2 += operator + " " + num2.rjust(width - 2) + "    "
+        line3 += "-" * width + "    "
+
+        if show_answers:
+            if operator == '+':
+                answer = str(int(num1) + int(num2))
+            else:
+                answer = str(int(num1) - int(num2))
+            line4 += answer.rjust(width) + "    "
+
+    arranged_problems = line1.rstrip() + "\n" + line2.rstrip() + "\n" + line3.rstrip()
+  
+    if show_answers:
+        arranged_problems += "\n" + line4.rstrip()
+
+    return arranged_problems

main.py

@@ -0,0 +1,10 @@
+# This entrypoint file to be used in development. Start by reading README.md
+from arithmetic_arranger import arithmetic_arranger
+from unittest import main
+
+
+print(arithmetic_arranger(["32 + 698", "3801 - 2", "45 + 43", "123 + 49"]))
+
+
+# Run unit tests automatically
+main(module='test_module', exit=False)

test_module.py

@@ -0,0 +1,43 @@
+import unittest
+from arithmetic_arranger import arithmetic_arranger
+
+
+# the test case
+class UnitTests(unittest.TestCase):
+    def test_arrangement(self):
+        actual = arithmetic_arranger(["3 + 855", "3801 - 2", "45 + 43", "123 + 49"])
+        expected = "    3      3801      45      123\n+ 855    -    2    + 43    +  49\n-----    ------    ----    -----"
+        self.assertEqual(actual, expected, 'Expected different output when calling "arithmetic_arranger()" with ["3 + 855", "3801 - 2", "45 + 43", "123 + 49"]')
+
+        actual = arithmetic_arranger(["11 + 4", "3801 - 2999", "1 + 2", "123 + 49", "1 - 9380"])
+        expected = "  11      3801      1      123         1\n+  4    - 2999    + 2    +  49    - 9380\n----    ------    ---    -----    ------"
+        self.assertEqual(actual, expected, 'Expected different output when calling "arithmetic_arranger()" with ["11 + 4", "3801 - 2999", "1 + 2", "123 + 49", "1 - 9380"]')
+
+    def test_too_many_problems(self):
+        actual = arithmetic_arranger(["44 + 815", "909 - 2", "45 + 43", "123 + 49", "888 + 40", "653 + 87"])
+        expected = "Error: Too many problems."
+        self.assertEqual(actual, expected, 'Expected calling "arithmetic_arranger()" with more than five problems to return "Error: Too many problems."')
+
+    def test_incorrect_operator(self):
+        actual = arithmetic_arranger(["3 / 855", "3801 - 2", "45 + 43", "123 + 49"])
+        expected = "Error: Operator must be '+' or '-'."
+        self.assertEqual(actual, expected, '''Expected calling "arithmetic_arranger()" with a problem that uses the "/" operator to return "Error: Operator must be '+' or '-'."''')
+        
+    def test_too_many_digits(self):
+        actual = arithmetic_arranger(["24 + 85215", "3801 - 2", "45 + 43", "123 + 49"])
+        expected = "Error: Numbers cannot be more than four digits."
+        self.assertEqual(actual, expected, 'Expected calling "arithmetic_arranger()" with a problem that has a number over 4 digits long to return "Error: Numbers cannot be more than four digits."')
+
+    def test_only_digits(self):
+        actual = arithmetic_arranger(["98 + 3g5", "3801 - 2", "45 + 43", "123 + 49"])
+        expected = "Error: Numbers must only contain digits."
+        self.assertEqual(actual, expected, 'Expected calling "arithmetic_arranger()" with a problem that contains a letter character in the number to return "Error: Numbers must only contain digits."')
+
+    def test_solutions(self):
+        actual = arithmetic_arranger(["32 - 698", "1 - 3801", "45 + 43", "123 + 49"], True)
+        expected = "   32         1      45      123\n- 698    - 3801    + 43    +  49\n-----    ------    ----    -----\n -666     -3800      88      172"
+        self.assertEqual(actual, expected, 'Expected solutions to be correctly displayed in output when calling "arithmetic_arranger()" with arithmetic problems and a second argument of `True`.')
+
+
+if __name__ == "__main__":
+    unittest.main()