View GitHub Repository average-minimum-distance

Schedule Inspection
Subscribe

conftest.reference_data()   A
last analyzed

↳ Parent: conftest

Complexity

Conditions 4

Size

Total Lines 14
Code Lines 12

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
eloc 12
dl 0
loc 14
rs 9.8
c 0
b 0
f 0
cc 4
nop 1
1 
import os
2 
import pickle
3 
import pytest
4 
import pathlib
5 

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    6
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                
@pytest.fixture(scope="session", autouse=True)

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                
def data_dir():

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    9
                

                                    
                                                    
                
                
                
    return pathlib.Path(__file__).absolute().parent / "data"

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    10
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                
@pytest.fixture(scope="session", autouse=True)

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    13
                

                                    
                                                    
                
                
                
def ccdc_enabled():

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    14
                

                                    
                                                    
                
                
                
    try:

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    15
                

                                    
                                                    
                
                
                
        import ccdc

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    17
                

                                    
                                                    
                
                
                
        return True

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    18
                

                                    
                                                    
                
                
                
    except Exception:

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    19
                

                                    
                                                    
                
                
                
        return False

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    20
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    21
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    22
                

                                    
                                                    
                
                
                
@pytest.fixture(scope="session", autouse=True)

            


            
                                                        
            
                                    
            
            

                
                    23
                

                                    
                                                    
                
                
                
def reference_data(data_dir):

            


            
                                                        
            
                                    
            
            

                
                    24
                

                                    
                                                    
                
                
                
    """Data fixture of amd.PeriodicSet objects to use in tests."""

            


            
                                                        
            
                                    
            
            

                
                    25
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    26
                

                                    
                                                    
                
                
                
    filenames = ["cubic", "T2_experimental", "CSD_families"]

            


            
                                                        
            
                                    
            
            

                
                    27
                

                                    
                                                    
                
                
                
    refs = {}

            


            
                                                        
            
                                    
            
            

                
                    28
                

                                    
                                                    
                
                
                
    for name in filenames:

            


            
                                                        
            
                                    
            
            

                
                    29
                

                                    
                                                    
                
                
                
        path = str(data_dir / f"{name}.pkl")

            


            
                                                        
            
                                    
            
            

                
                    30
                

                                    
                                                    
                
                
                
        with open(path, "rb") as f:

            


            
                                                        
            
                                    
            
            

                
                    31
                

                                    
                                                    
                
                
                
            data = pickle.load(f)

            


            
                                                        
            
                                    
            
            

                
                    32
                

                                    
                                                    
                
                
                
        if not data:

            


            
                                                        
            
                                    
            
            

                
                    33
                

                                    
                                                    
                
                
                
            raise ValueError(f"Data not found in path {path}")

            


            
                                                        
            
                                    
            
            

                
                    34
                

                                    
                                                    
                
                
                
        refs[name] = data

            


            
                                                        
            
                                    
            
            

                
                    35
                

                                    
                                                    
                
                
                
    return refs

            


            
                                                        
            
                                    
            
            

                
                    36