Add analysis_pipeline to for the local test as well

KosinskiLab · Sep 25, 2024 · 9ae04e2 · 9ae04e2
1 parent 31a42e6
commit 9ae04e2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_get_good_inter_pae.py b/test/test_get_good_inter_pae.py
@@ -1,12 +1,13 @@
 from absl.testing import absltest
 from unittest.mock import patch, mock_open, MagicMock
-import json
+import os, sys, json
 import pandas as pd
+# Add the root directory of your project to sys.path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../alphapulldown/analysis_pipeline')))
 from alphapulldown.analysis_pipeline.get_good_inter_pae import obtain_seq_lengths, main
 
 
 class TestGetGoodInterPae(absltest.TestCase):
-
     @patch('alphapulldown.analysis_pipeline.get_good_inter_pae.os.path.exists')
     @patch('alphapulldown.analysis_pipeline.get_good_inter_pae.PDBParser.get_structure')
     @patch('alphapulldown.analysis_pipeline.get_good_inter_pae.PPBuilder.build_peptides')