|
1 | 1 | import os |
| 2 | +import pytest |
2 | 3 |
|
3 | 4 | from alphafold3_pytorch import collate_inputs_to_batched_atom_input |
4 | 5 | from alphafold3_pytorch.alphafold3 import Alphafold3 |
|
11 | 12 |
|
12 | 13 |
|
13 | 14 | def test_data_input(): |
| 15 | + pytest.skip(f"data/mmcif not populated yet") |
| 16 | + |
| 17 | + data_test = os.path.join("data", "test") |
| 18 | + |
14 | 19 | """Test a PDBDataset constructed using a WeightedPDBSampler.""" |
15 | | - interface_mapping_path = os.path.join("data", "test", "interface_cluster_mapping.csv") |
| 20 | + interface_mapping_path = os.path.join(data_test, "interface_cluster_mapping.csv") |
16 | 21 | chain_mapping_paths = [ |
17 | | - os.path.join("data", "test", "ligand_chain_cluster_mapping.csv"), |
18 | | - os.path.join( |
19 | | - "data", |
20 | | - "test", |
21 | | - "nucleic_acid_chain_cluster_mapping.csv", |
22 | | - ), |
23 | | - os.path.join("data", "test", "peptide_chain_cluster_mapping.csv"), |
24 | | - os.path.join("data", "test", "protein_chain_cluster_mapping.csv"), |
| 22 | + os.path.join(data_test, "ligand_chain_cluster_mapping.csv"), |
| 23 | + os.path.join(data_test, "nucleic_acid_chain_cluster_mapping.csv"), |
| 24 | + os.path.join(data_test, "peptide_chain_cluster_mapping.csv"), |
| 25 | + os.path.join(data_test, "protein_chain_cluster_mapping.csv"), |
25 | 26 | ] |
26 | 27 |
|
27 | 28 | sampler = WeightedPDBSampler( |
|
0 commit comments