From bfb62acb355a50b6e067604724c425ee1e14af7f Mon Sep 17 00:00:00 2001
From: Blesa Patricia <blesapati@gmail.com>
Date: Tue, 18 Apr 2017 10:37:45 +0200
Subject: [PATCH] feat(test) create 3 test cases and modified function to pass
 them

---
 oddavg/odd_avg.py | 13 ++++++++-----
 oddavg/test.py    | 19 +++++++++++++++++++
 2 files changed, 27 insertions(+), 5 deletions(-)
 create mode 100644 oddavg/test.py

diff --git a/oddavg/odd_avg.py b/oddavg/odd_avg.py
index 24fc178..665f3e2 100644
--- a/oddavg/odd_avg.py
+++ b/oddavg/odd_avg.py
@@ -1,15 +1,18 @@
 # Create a function called `odd_average` that takes a list of numbers as parameter
 # and returns the average value of the odd numbers in the list
 # Create basic unit tests for it with at least 3 different test cases
-nums = [15, 2, 3, 4, 3, 6]
+nums = [1, 2, 3, 4, 5]
 
 def odd_average(nums):
     odds = 0
     odd_num = []
     for num in nums:
-        if num%2 == 1:
-            odds += num
-            odd_num.append(num)
-    return odds//len(odd_num)
+        try:
+            if num%2 == 1:
+                odds += num
+                odd_num.append(num)
+            return odds//len(odd_num)
+        except ZeroDivisionError:
+            return 0
 
 print(odd_average(nums))
diff --git a/oddavg/test.py b/oddavg/test.py
new file mode 100644
index 0000000..85d4209
--- /dev/null
+++ b/oddavg/test.py
@@ -0,0 +1,19 @@
+import unittest
+from odd_avg import odd_average
+
+class AverageMethod(unittest.TestCase):
+
+    def test_average(self):
+        list_of_nums = [9, 9, 10, 9, 3]
+        self.assertEqual(odd_average(list_of_nums), 9)
+
+    def test_odds(self):
+        list_of_nums = [4, 6, 8]
+        self.assertEqual(odd_average(list_of_nums), 0)
+
+    def test_empty_list(self):
+        list_of_nums = [ ]
+        self.assertEqual(odd_average(list_of_nums), None)
+
+if __name__ == '__main__':
+    unittest.main()