from __future__ import print_function
import pytest
from threeML import *
from .conftest import get_grb_model
try:
import ROOT
except:
has_root = False
else:
has_root = True
skip_if_ROOT_is_available = pytest.mark.skipif(
has_root, reason="ROOT is available. Skipping incompatible tests."
)
# Define a dummy function to return always the same model
[docs]
def get_model(id):
return get_grb_model(Powerlaw())
[docs]
def test_joint_likelihood_set(data_list_bn090217206_nai6):
def get_data(id):
return data_list_bn090217206_nai6
jlset = JointLikelihoodSet(
data_getter=get_data, model_getter=get_model, n_iterations=10
)
jlset.go(compute_covariance=False)
[docs]
@skip_if_ROOT_is_available
def test_joint_likelihood_set_parallel(data_list_bn090217206_nai6):
def get_data(id):
return data_list_bn090217206_nai6
jlset = JointLikelihoodSet(
data_getter=get_data, model_getter=get_model, n_iterations=10
)
with parallel_computation(start_cluster=False):
res = jlset.go(compute_covariance=False)
print(res)