test : reformat with precommit

Author: aditya0by0

tests/unit/dataset_classes/testProteinPretrainingData.py

@@ -1,5 +1,6 @@
 import unittest
 from unittest.mock import PropertyMock, mock_open, patch
+
 from chebai.preprocessing.datasets.protein_pretraining import _ProteinPretrainingData
 from chebai.preprocessing.reader import ProteinDataReader
 from tests.unit.mock_data.ontology_mock_data import GOUniProtMockData
@@ -41,7 +42,9 @@ def setUpClass(
         new_callable=mock_open,
         read_data=GOUniProtMockData.get_UniProt_raw_data(),
     )
-    def test_parse_protein_data_for_pretraining(self, mock_open_file: mock_open) -> None:
+    def test_parse_protein_data_for_pretraining(
+        self, mock_open_file: mock_open
+    ) -> None:
         """
         Tests the _parse_protein_data_for_pretraining method.
 
@@ -55,15 +58,15 @@ def test_parse_protein_data_for_pretraining(self, mock_open_file: mock_open) ->
 
         # Assert that all expected Swiss-Prot IDs are present in the DataFrame
         self.assertEqual(
-            set(pretrain_df['swiss_id']),
+            set(pretrain_df["swiss_id"]),
             set(list_of_pretrain_swiss_ids),
-            msg="The parsed DataFrame does not contain the expected Swiss-Prot IDs for pretraining."
+            msg="The parsed DataFrame does not contain the expected Swiss-Prot IDs for pretraining.",
         )
 
         # Assert that all sequences are not empty
         self.assertTrue(
-            pretrain_df['sequence'].str.len().gt(0).all(),
-            msg="Some protein sequences in the pretraining DataFrame are empty."
+            pretrain_df["sequence"].str.len().gt(0).all(),
+            msg="Some protein sequences in the pretraining DataFrame are empty.",
         )
 
 

tests/unit/mock_data/ontology_mock_data.py

@@ -751,19 +751,19 @@ def get_UniProt_raw_data() -> str:
             "SQ   SEQUENCE   60 AA;  29735 MW;  B4840739BF7D4121 CRC64;\n"
             "     XAFSAEDVLK EYDRRRRMEA LLLSLYYPND RKLLDYKEWS PPRVQVECPK APVEWNNPPS\n"
             "//\n"
-            # Below protein with a `valid` associated GO class but without sequence string 
+            # Below protein with a `valid` associated GO class but without sequence string
             "ID   Swiss_Prot_10              Reviewed;         60 AA.\n"
             "AC   Q6GZX4;\n"
             "DR   GO; GO:0000027; P:regulation of viral transcription; EXP:InterPro.\n"
             "SQ   SEQUENCE   60 AA;  29735 MW;  B4840739BF7D4121 CRC64;\n"
-            "     \n"  
+            "     \n"
             "//\n"
-            # Below protein with a `Invalid` associated GO class but without sequence string 
+            # Below protein with a `Invalid` associated GO class but without sequence string
             "ID   Swiss_Prot_11              Reviewed;         60 AA.\n"
             "AC   Q6GZX4;\n"
             "DR   GO; GO:0000028; P:regulation of viral transcription; ND:NoData.\n"
             "SQ   SEQUENCE   60 AA;  29735 MW;  B4840739BF7D4121 CRC64;\n"
-            "     \n"  
+            "     \n"
             "//\n"
         )