Fixed the tests to check the entire dataframe rather than specific cells

Author: marcoagonzales007

process_report/tests/unit/processors/test_add_institution_processor.py

@@ -6,14 +6,14 @@
 
 
 class TestAddInstitutionProcessor(TestCase):
-    def _get_test_data(self, pi_names, projects=None):
-        if projects is None:
-            projects = [f"Project{i}" for i in range(len(pi_names))]
+    def _get_test_data(self, pi_names, projects=None, institutions=None):
+        if institutions is None:
+            institutions = ["" for i in pi_names]
         return pandas.DataFrame(
             {
                 "Manager (PI)": pi_names,
                 "Project - Allocation": projects,
-                "Institution": ["" for _ in pi_names],
+                "Institution": institutions,
             }
         )
 
@@ -35,8 +35,12 @@ def test_add_institution(self, mock_load_institute_list):
         processor = test_utils.new_add_institution_processor(data=test_data)
         processor.process()
         output = processor.data
-        assert output.loc[0, "Institution"] == "Boston University"
-        assert output.loc[1, "Institution"] == "MIT"
+        answer_data = self._get_test_data(
+            pi_names=["pi1@bu.edu", "pi2@mit.edu"],
+            projects=["ProjectA", "ProjectB"],
+            institutions=["Boston University", "MIT"],
+        )
+        assert output.equals(answer_data)
 
     @mock.patch("process_report.util.load_institute_list")
     def test_add_institution_missing_pi(self, mock_load_institute_list):