update testcase as per transitive go ids

Author: aditya0by0

tests/unit/dataset_classes/testGOUniProDataExtractor.py

@@ -1,4 +1,5 @@
 import unittest
+from collections import OrderedDict
 from unittest.mock import PropertyMock, mock_open, patch
 
 import fastobo
@@ -141,7 +142,14 @@ def test_get_swiss_to_go_mapping(self, mock_open) -> None:
         Test the extraction of SwissProt to GO term mapping.
         """
         mapping_df = self.extractor._get_swiss_to_go_mapping()
-        expected_df = GOUniProtMockData.get_data_in_dataframe().iloc[:, :4]
+        expected_df = pd.DataFrame(
+            OrderedDict(
+                swiss_id=["Swiss_Prot_1", "Swiss_Prot_2"],
+                accession=["Q6GZX4", "DCGZX4"],
+                go_ids=[[2, 3, 5], [2, 5]],
+                sequence=list(GOUniProtMockData.protein_sequences().values()),
+            )
+        )
 
         pd.testing.assert_frame_equal(
             mapping_df,

tests/unit/mock_data/ontology_mock_data.py

@@ -736,11 +736,11 @@ def get_data_in_dataframe() -> pd.DataFrame:
         expected_data = OrderedDict(
             swiss_id=["Swiss_Prot_1", "Swiss_Prot_2"],
             accession=["Q6GZX4", "DCGZX4"],
-            go_ids=[[2, 3, 5], [2, 5]],
+            go_ids=[[1, 2, 3, 5], [1, 2, 5]],
             sequence=list(GOUniProtMockData.protein_sequences().values()),
             **{
                 #   SP_1,  SP_2
-                1: [False, False],
+                1: [True, True],
                 2: [True, True],
                 3: [True, False],
                 4: [False, False],