test_mcmc_utils

Module Contents

Classes

TestMCMCUtils A suite of tests for the h0rton.h0_inference.mcmc_utils package
class test_mcmc_utils.TestMCMCUtils(methodName='runTest')[source]

Bases: unittest.TestCase

A suite of tests for the h0rton.h0_inference.mcmc_utils package

classmethod setUpClass(cls)[source]

Hook method for setting up class fixture before running tests in the class.

test_get_lens_kwargs(self)[source]

Test if the default lens kwargs dictionary config is plausible

test_get_ps_kwargs(self)[source]

Test if the default ps kwargs dictionary config defined on the image plane is plausible

test_get_ps_kwargs_src_plane(self)[source]

Test if the default ps kwargs dictionary config defined on the source plane is plausible

test_get_light_kwargs(self)[source]

Test if the default light kwargs dictionary config is plausible

test_get_special_kwargs(self)[source]

Test if the default special kwargs dictionary config is plausible

test_postprocess_mcmc_chain(self)[source]
test_HybridBNNPenalty(self)[source]
test_get_idx_for_params(self)[source]

Test if get_idx_for_params returns the right indices

test_remove_parameters_from_pred(self)[source]

Test if correct parameters are removed from the NN output

test_split_component_param(self)[source]

Test if string split of Baobab column names is done correctly

test_dict_to_array(self)[source]

Test if dict from MCMC iteration is converted into array form correctly

test_reorder_to_param_class(self)[source]

Test if dict from MCMC iteration is converted into array form correctly