Add test case to schema_simple for three part make function

Author: ethho

tests/conftest.py

@@ -336,6 +336,7 @@ def schema_simp(connection_test, prefix):
     schema(schema_simple.B)
     schema(schema_simple.L)
     schema(schema_simple.D)
+    schema(schema_simple.N)
     schema(schema_simple.E)
     schema(schema_simple.F)
     schema(schema_simple.F)

tests/schema_simple.py

@@ -92,7 +92,33 @@ def _make_tuples(self, key):
         # make reference to a random tuple from L
         random.seed(str(key))
         lookup = list(L().fetch("KEY"))
-        self.insert(dict(key, id_d=i, **random.choice(lookup)) for i in range(4))
+        self.insert(dict(key, id_d=i, **random.choice(lookup))
+                    for i in range(4))
+
+
+class N(dj.Computed):
+    definition = """
+    # test for three part make function
+    -> A
+    id_d :int
+    ---
+    -> L
+    """
+
+    def make_fetch(self, key):
+        # make reference to a random tuple from L
+        lookup = list(L().fetch("KEY"))
+        inputs = key, lookup
+        return inputs
+
+    def make_compute(self, inputs):
+        key, lookup = inputs
+        random.seed(str(key))
+        values = [dict(key, id_d=i, **random.choice(lookup)) for i in range(4)]
+        return values
+
+    def make_insert(self, values):
+        self.insert(values)
 
 
 class E(dj.Computed):