INIT example

ShMcK · ShMcK · commit a51b0f789953 · 2020-06-14T11:47:39.000-07:00
diff --git a/coderoad-starter/src/__init__.py b/coderoad-starter/src/__init__.py
@@ -1,2 +1 @@
-from math import Add
-from math import Subtract
+from .example import add
diff --git a/coderoad-starter/src/example.py b/coderoad-starter/src/example.py
@@ -0,0 +1,15 @@
+def add(*args):
+    '''Add 1 or more numbers together'''
+    sum = 0
+    for arg in args:
+        sum += arg
+    return sum
+
+def main():
+    print('hello')
+    print(str(add(1)))
+    print(str(add(1, 2)))
+    print(str(add(1, 2, 3)))
+
+if __name__ == '__main__':
+    main()
diff --git a/coderoad-starter/src/math.py b/coderoad-starter/src/math.py
diff --git a/coderoad-starter/tests/context.py b/coderoad-starter/tests/context.py
@@ -3,4 +3,4 @@
 sys.path.insert(0, os.path.abspath(
     os.path.join(os.path.dirname(__file__), '..')))
 
-import src
+import src
diff --git a/coderoad-starter/tests/math_test.py b/coderoad-starter/tests/math_test.py
@@ -0,0 +1,15 @@
+import unittest
+from context import src
+
+class MathTest(unittest.TestCase):
+    def test_add_no_numbers(self):
+        self.assertEqual(src.add(), 0, "Should return 0 with no params")
+    def test_add_one_number(self):
+        self.assertEqual(src.add(1), 1, "Should add one number to 0")
+    def test_add_two_numbers(self):
+        self.assertEqual(src.add(1, 2), 3, "Should add two numbers together")
+    def test_add_three_numbers(self):
+        self.assertEqual(src.add(1, 2, 3), 6, "Should add three numbers together")
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/coderoad-starter/tests/mock_test.py b/coderoad-starter/tests/mock_test.py

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1 @@`
`1`		`-from math import Add`
`2`		`-from math import Subtract`
	`1`	`+from .example import add`