update unit test

Author: aditya0by0

tests/unit/dataset_classes/testGOUniProDataExtractor.py

@@ -6,8 +6,10 @@
 import networkx as nx
 import pandas as pd
 
-from chebai.preprocessing.datasets.deepGO.go_uniprot import _GOUniProtDataExtractor
-from chebai.preprocessing.reader import ProteinDataReader
+from chebai_proteins.preprocessing.datasets.deepGO.go_uniprot import (
+    _GOUniProtDataExtractor,
+)
+from chebai_proteins.preprocessing.reader import ProteinDataReader
 from tests.unit.mock_data.ontology_mock_data import GOUniProtMockData
 
 

tests/unit/dataset_classes/testGoUniProtOverX.py

@@ -5,7 +5,7 @@
 import networkx as nx
 import pandas as pd
 
-from chebai.preprocessing.datasets.deepGO.go_uniprot import _GOUniProtOverX
+from chebai_proteins.preprocessing.datasets.deepGO.go_uniprot import _GOUniProtOverX
 from tests.unit.mock_data.ontology_mock_data import GOUniProtMockData
 
 

tests/unit/dataset_classes/testProteinPretrainingData.py

@@ -1,10 +1,10 @@
 import unittest
 from unittest.mock import PropertyMock, mock_open, patch
 
-from chebai.preprocessing.datasets.deepGO.protein_pretraining import (
+from chebai_proteins.preprocessing.datasets.deepGO.protein_pretraining import (
     _ProteinPretrainingData,
 )
-from chebai.preprocessing.reader import ProteinDataReader
+from chebai_proteins.preprocessing.reader import ProteinDataReader
 from tests.unit.mock_data.ontology_mock_data import GOUniProtMockData
 
 

tests/unit/readers/testProteinDataReader.py

@@ -2,7 +2,7 @@
 from typing import List
 from unittest.mock import mock_open, patch
 
-from chebai.preprocessing.reader import EMBEDDING_OFFSET, ProteinDataReader
+from chebai_proteins.preprocessing.reader import EMBEDDING_OFFSET, ProteinDataReader
 
 
 class TestProteinDataReader(unittest.TestCase):