feat(test) create 3 test cases and modified function to pass them

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))

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()