add a test for configs

Author: lucidrains

tests/alphafold3.yaml

@@ -0,0 +1,23 @@
+---
+dim_atom_inputs: 10
+dim_template_feats: 10
+dim_template_model: 64
+atoms_per_window: 27
+dim_atom: 128
+dim_atompair_inputs: 5
+dim_atompair: 16
+dim_input_embedder_token: 384
+dim_single: 384
+dim_pairwise: 128
+dim_token: 768
+distance_bins: 38
+ignore_index: -1
+num_dist_bins: null
+num_plddt_bins: 50
+num_pde_bins: 64
+num_pae_bins: 64
+sigma_data: 16
+diffusion_num_augmentations: 4
+loss_confidence_weight: 0.0001
+loss_distogram_weight: 0.01
+loss_diffusion_weight: 4.

tests/test_config.py

@@ -0,0 +1,24 @@
+import os
+os.environ['TYPECHECK'] = 'True'
+
+import torch
+import pytest
+from pathlib import Path
+
+from alphafold3_pytorch.alphafold3 import Alphafold3
+
+from alphafold3_pytorch.configs import (
+    Alphafold3Config
+)
+
+# constants
+
+curr_dir = Path(__file__).parents[0]
+
+# tests
+
+def test_alphafold3_config():
+    af3_yaml = curr_dir / 'alphafold3.yaml'
+
+    alphafold3 = Alphafold3Config.create_instance_from_yaml_file(af3_yaml)
+    assert isinstance(alphafold3, Alphafold3)