Add extra tests

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = undefined_AD
-version = 2.0.4
+version = 2.1.0
 author = cs107-undefined
 author_email = [email protected]
 description = Auto differetiation library

src/undefined_AD.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: undefined-AD
-Version: 2.0.4
+Version: 2.1.0
 Summary: Auto differetiation library
 Home-page: https://github.com/cs107-undefined/cs107-FinalProject
 Author: cs107-undefined
@@ -17,9 +17,11 @@ License-File: LICENSE
 
 # cs107-FinalProject
 
-[![Build Status](https://app.travis-ci.com/cs107-undefined/cs107-FinalProject.svg?branch=milestone2)](https://app.travis-ci.com/cs107-undefined/cs107-FinalProject)
+[![Build Status](https://app.travis-ci.com/cs107-undefined/cs107-FinalProject.svg?branch=final_milestone)](https://app.travis-ci.com/cs107-undefined/cs107-FinalProject)
 
-[![codecov](https://codecov.io/gh/cs107-undefined/cs107-FinalProject/branch/milestone2/graph/badge.svg?token=MWEZONI94C)](https://codecov.io/gh/cs107-undefined/cs107-FinalProject)
+[![codecov](https://codecov.io/gh/cs107-undefined/cs107-FinalProject/branch/final_milestone/graph/badge.svg?token=MWEZONI94C)](https://codecov.io/gh/cs107-undefined/cs107-FinalProject)
+
+[![Documentation Status](https://readthedocs.org/projects/cs107-undefined/badge/?version=latest)](https://cs107-undefined.readthedocs.io/en/latest/?badge=latest)
 
 [**DOCUMENTATION**](https://cs107-undefined.readthedocs.io/en/latest/)
 

test/test_API.py

@@ -67,7 +67,46 @@ def test_trace_with_incompatible_seeds(self):
         with self.assertRaises(AttributeError):
             trace(self.f2, seeds = np.array([[1,3,2],[1,3,2],[2,3,3]]), x=np.array(
                 [[2, 3]]), y=np.array([[1, 2]]))
-    #def test_trace_with_vector_inputs_with_seeds(self):
+
+    def test_trace_with_incompatible_seeds_reverse(self):
+        with self.assertRaises(AttributeError):
+            trace(self.f2, seeds = np.array([[1,3,2],[1,3,2],[2,3,3]]), mode = "reverse", x=np.array(
+                [[2, 3]]), y=np.array([[1, 2]]))
+
+    def test_trace_with_scalar_inputs_seeds(self):
+        with self.assertRaises(TypeError):
+            trace(self.f1, seeds = np.array([[1,0]]), x=1)
+        self.assertEqual(trace(self.f1, seeds = 2, x=1), (3,2))
+        self.assertEqual(trace(self.f2, seeds = np.array([[1,0],[0,1]]), x=2, y=999)[1], [[1], [1]])
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = 1, x=2, y=999)
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = "seed", x=2, y=999)
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = np.array([[1,0],[0,1]]), x="2", y=999)
+        with self.assertRaises(AttributeError):
+            trace(self.f2, seeds = np.array([[1],[0]]), x=2, y=999)
+
+    def test_trace_with_scalar_inputs_seeds_reverse(self):
+        with self.assertRaises(TypeError):
+            trace(self.f1, seeds = np.array([[1,0]]), mode='reverse', x=1)
+        self.assertEqual(trace(self.f1, seeds = 2, mode='reverse', x=1), (3,2))
+        self.assertEqual(trace(self.f2, seeds = np.array([[1,0],[0,1]]), mode='reverse', x=2, y=999)[1], [[1], [1]])
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = 1, mode='reverse', x=2, y=999)
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = "seed", mode='reverse', x=2, y=999)
+        with self.assertRaises(TypeError):
+            trace(self.f2, seeds = np.array([[1,0],[0,1]]), mode='reverse', x="2", y=999)
+        with self.assertRaises(AttributeError):
+            trace(self.f2, seeds = np.array([[1],[0]]), mode='reverse', x=2, y=999)
+
+    def test_trace_with_vector_inputs_seeds(self):
+        self.assertEqual(trace(self.f2, seeds = np.array([[1,0],[0,1]]), x=np.array([[2, 3]]), y=np.array([[1, 2]]))[1], [[1, 1], [1, 1]])
+
+    def test_trace_with_vector_inputs_seeds_reverse(self):
+        self.assertEqual(trace(self.f2, seeds = np.array([[1,0],[0,1]]), mode='reverse', x=np.array([[2, 3]]), y=np.array([[1, 2]]))[1], [[1, 1], [1, 1]])
+
     def test_trace_with_different_moded(self):
         self.assertEqual(trace(self.f1, x=2), (4, 1))
         self.assertEqual(trace(self.f1, mode='forward', x=2), (4, 1))
@@ -111,10 +150,18 @@ def test_trace_multiple_vector_inputs(self):
         self.assertEqual(trace(self.f3, mode='reverse', x=np.array(
             [[2, 2]]), y=np.array([[4, 4]]))[1], [[2., 2.], [0.25, 0.25]])
 
-    def test_trace_single_vector_inputs(self):
+    def test_trace_single_vector_input(self):
         self.assertEqual(trace(self.f1, x=np.array([[2, 2]]))[1], [[1, 1]])
+        with self.assertRaises(TypeError):
+            trace(self.f1, x=np.array([]))
 
+    def test_trace_non_lambda_function(self):
+        with self.assertRaises(TypeError):
+            trace("Function", x = 1)
 
+    def test_trace_vector_functions(self):
+        self.assertEqual(trace([self.f2,self.f3], x = 2, y = 4)[1].tolist(), [[[1.0], [1.0]], [[2.0], [0.25]]])
+
     def test_mixed_inputs(self):
         self.assertEqual(trace(self.f3, x=np.array([[2, 2]]), y=4)[
                          0].tolist(), [[6, 6]])
@@ -125,7 +172,8 @@ def test_mixed_inputs(self):
                          1], [[2., 2.], [0.25, 0.25]])
         self.assertEqual(trace(self.f3, mode='reverse', x=np.array([[2, 2]]), y=4)[
                          1], [[2., 2.], [0.25, 0.25]])
-
+        self.assertEqual(trace(self.f3, mode='reverse', x=np.array([[2, 2]]), y=np.array([[4, 4]]))[
+                         1], [[2., 2.], [0.25, 0.25]])
 
 if __name__ == "__main__":
     unittest.main()

test/test_graph_generator_operation.py

@@ -381,9 +381,6 @@ def test_ge(self):
 		with self.assertRaises(TypeError):
 			self.f25 >= "self.f19"
 
-
-
-
 
 # this will help to run the unittest directly. 
 if __name__ == "__main__":