Python NestedSampler.setup_outputの例

コード例 #1

ファイル: test_general_config.py プロジェクト: mj-will/nessai

def test_setup_output_w_resume(sampler, tmpdir):
    """Test output configuration with a specified resume file"""
    p = tmpdir.mkdir('outputs')
    sampler.plot = False
    resume_file = \
        NestedSampler.setup_output(sampler, f'{p}/tests', 'resume.pkl')
    assert resume_file == f'{p}/tests/resume.pkl'

コード例 #2

ファイル: test_general_config.py プロジェクト: mj-will/nessai

def test_setup_output(sampler, tmpdir):
    """Test setting up the output directories"""
    p = tmpdir.mkdir('outputs')
    sampler.plot = False
    resume_file = NestedSampler.setup_output(sampler, f'{p}/tests')
    assert os.path.exists(f'{p}/tests')
    assert resume_file == f'{p}/tests/nested_sampler_resume.pkl'

コード例 #3

ファイル: test_general_config.py プロジェクト: mj-will/nessai

def test_setup_output_w_plotting(sampler, tmpdir):
    """Test setting up the output directories with plot=True"""
    p = tmpdir.mkdir('outputs')
    sampler.plot = True
    NestedSampler.setup_output(sampler, f'{p}/tests')
    assert os.path.exists(f'{p}/tests/diagnostics')