File size: 545 Bytes
80a72c3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from pathlib import Path
from model.models.results import PredictionResult

def test_results_without_chain_id():

    mock_pdb_path = Path("/path/to/test_chain_id.pdb")
    mock_chain_id = "test_chain_id"
    mock_md5 = "1234567890abcdef"
    mock_chopping = "12-34_56-78"
    mock_confidence = 0.0123

    res = PredictionResult(
        pdb_path=mock_pdb_path,
        sequence_md5=mock_md5,
        nres=1234,
        ndom=1,
        chopping=mock_chopping,
        confidence=mock_confidence,
    )

    assert res.chain_id == mock_chain_id