bug fixes and updates to code

verarbeitung/construct_new_graph/add_citations_rec.py

@@ -22,6 +22,22 @@ sys.path.append("../")
 from input.publication import Publication
 from verarbeitung.get_pub_from_input import get_pub
 
+def create_graph_structure_citations_test(pub, search_depth, search_depth_max, cit_type, test_var, test_nodes, test_edges):
+    '''
+        :param test_nodes:      list of publications from unit test
+        :type test_nodes:       List[Publication]
+
+        :param test_edges:      list of links from unit test
+        :type test_edges:       List[List[String,String]]
+
+        for unit test purposes only
+    '''
+    global nodes, edges
+    nodes = test_nodes
+    edges = test_edges
+    return(nodes, edges, create_graph_structure_citations(pub, search_depth, search_depth_max, cit_type, test_var))
+
+
 def get_cit_type_list(pub, cit_type):
     '''
         :param pub:                 Publication which citations will be added
@@ -32,10 +48,12 @@ def get_cit_type_list(pub, cit_type):
 
         function to create nodes and edges and call create_graph_structure_citations
     '''
-    if (cit_type == "Citation"):
+    if cit_type == "Citation":
         return(pub.citations)
-    else:
+    elif cit_type == "Reference":
         return(pub.references)
+    else:
+        return(ValueError)
 
 def create_global_lists_cit(input_nodes, input_edges, pub, search_depth, search_depth_max, cit_type, test_var):
     '''