Conditions | 3 |
Total Lines | 15 |
Code Lines | 11 |
Lines | 0 |
Ratio | 0 % |
Changes | 0 |
1 | import pytest |
||
11 | @pytest.fixture(scope='session', autouse=True) |
||
12 | def reference_data(root_dir): |
||
13 | |||
14 | filenames = { # .pkl files with PeriodicSets. |
||
15 | 'cubic': 'cubic', |
||
16 | 'T2_experimental': 'T2_experimental', |
||
17 | 'CSD_families': 'CSD_families', |
||
18 | } |
||
19 | |||
20 | refs = {} |
||
21 | for name in filenames: |
||
22 | with open(os.path.join(root_dir, filenames[name] + '.pkl'), 'rb') as f: |
||
23 | refs[name] = pickle.load(f) |
||
24 | |||
25 | return refs |
||
26 |