From cfa96dd58f2f3279b4ddc0a25f3d6681d4d39bab Mon Sep 17 00:00:00 2001 From: Dima Molodenskiy Date: Thu, 5 Sep 2024 11:09:01 +0200 Subject: [PATCH] Clean pkl and json files before tests --- test/test_features_with_templates.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/test_features_with_templates.py b/test/test_features_with_templates.py index e0a66c8b..4975f39a 100644 --- a/test/test_features_with_templates.py +++ b/test/test_features_with_templates.py @@ -68,6 +68,10 @@ def run_features_generation(self, file_name, chain_id, file_extension, use_mmseq desc_file.write(f">{file_name}_{chain_id}, {file_name}.{file_extension}, {chain_id}\n") assert Path(f"{self.TEST_DATA_DIR}/fastas/{file_name}_{chain_id}.fasta").exists() + # Remove .pkl, .pkl.xz, .json, .json.xz files from the features directory + for ext in ['*.pkl', '*.pkl.xz', '*.json', '*.json.xz']: + for f in self.TEST_DATA_DIR.glob(f'features/{ext}'): + f.unlink() cmd = [ 'create_individual_features.py',