stable version and bug fixes for graph update

verarbeitung/test/update_graph_unittest.py

@@ -9,7 +9,7 @@ from verarbeitung.construct_new_graph.initialize_graph import init_graph_constru
 from verarbeitung.construct_new_graph.export_to_json import output_to_json
 from verarbeitung.update_graph.import_from_json import input_from_json
 from verarbeitung.update_graph.update_graph import update_graph, get_old_input_dois, get_new_input_dois
-from verarbeitung.update_graph.update_depth import reduce_max_height_depth_test, get_old_height_depth
+from verarbeitung.update_graph.update_depth import reduce_max_height_depth_test, get_old_max_references_citations_test
 from verarbeitung.get_pub_from_input import input_test_func
 
 class UpdatingTest(unittest.TestCase):
@@ -66,8 +66,8 @@ class UpdatingTest(unittest.TestCase):
 
      # hard to test because we only have dois as test objects and no urls variant
      def test_get_new_input_dois(self):
-          new_dois = ['pub_lg_2_i', 'pub_lg_1_i', 'pub_lg_c_i']
-          self.assertCountEqual(get_new_input_dois(new_dois, True), ['pub_lg_2_i', 'pub_lg_1_i', 'pub_lg_c_i'])
+          new_dois = ['doi_lg_2_i', 'doi_lg_1_i', 'doi_cg_i']
+          self.assertCountEqual(get_new_input_dois(new_dois, True), ['doi_lg_2_i', 'doi_lg_1_i', 'doi_cg_i'])
 
      
      # update_depth.py:
@@ -104,6 +104,38 @@ class UpdatingTest(unittest.TestCase):
           self.assertCountEqual(reduce_max_height_depth_test(pubs, 1, "Depth"), [pub_lg_2_i, pub_lg_2_d_11, pub_lg_2_h_11, pub_lg_2_h_21])
           self.assertCountEqual(reduce_max_height_depth_test(pubs, 0, "Depth"), [pub_lg_2_i, pub_lg_2_h_11, pub_lg_2_h_21])
 
+     def test_get_old_max_references(self):
+          pub_lg_2_i = input_test_func('doi_lg_2_i')
+          pub_lg_2_i.group = 0
+          pub_lg_2_h_11 = input_test_func('doi_lg_2_h11')
+          pub_lg_2_h_11.group = 1
+          pub_lg_2_d_11 = input_test_func('doi_lg_2_d11')
+          pub_lg_2_d_11.group = -1
+          pub_lg_2_h_21 = input_test_func('doi_lg_2_h21')
+          pub_lg_2_h_21.group = 2
+          pub_lg_2_d_21 = input_test_func('doi_lg_2_d21')
+          pub_lg_2_d_21.group = -2
+          pub_lg_2_d_22 = input_test_func('doi_lg_2_d22')
+          pub_lg_2_d_22.group = -2
+          pubs = [pub_lg_2_i, pub_lg_2_h_11, pub_lg_2_h_21, pub_lg_2_d_11, pub_lg_2_d_21, pub_lg_2_d_22]
+          self.assertCountEqual(get_old_max_references_citations_test(pubs, 2, "Depth"), [pub_lg_2_d_21,pub_lg_2_d_22])
+
+     def test_get_old_max_citations(self):
+          pub_lg_2_i = input_test_func('doi_lg_2_i')
+          pub_lg_2_i.group = 0
+          pub_lg_2_h_11 = input_test_func('doi_lg_2_h11')
+          pub_lg_2_h_11.group = 1
+          pub_lg_2_d_11 = input_test_func('doi_lg_2_d11')
+          pub_lg_2_d_11.group = -1
+          pub_lg_2_h_21 = input_test_func('doi_lg_2_h21')
+          pub_lg_2_h_21.group = 2
+          pub_lg_2_h_22 = input_test_func('doi_lg_2_h22')
+          pub_lg_2_h_22.group = 2
+          pub_lg_2_d_21 = input_test_func('doi_lg_2_d21')
+          pub_lg_2_d_21.group = -2
+          pubs = [pub_lg_2_i, pub_lg_2_h_11, pub_lg_2_h_21, pub_lg_2_h_22,pub_lg_2_d_11, pub_lg_2_d_21]
+          self.assertCountEqual(get_old_max_references_citations_test(pubs, 2, "Height"), [pub_lg_2_h_21,pub_lg_2_h_22])
+