import unittest from os.path import join from flowvb.core._lower_bound import _LowerBound from tests.test_old_faithful import makeTestFaithful TEST_ACCURACY = 3 MAX_DIFF = pow(10, -TEST_ACCURACY) TEST_DATA_LOC = join('tests', 'data', 'old_faithful') TestLogDirichletNormalizationPrior = makeTestFaithful( 'log_dirichlet_normalization_prior.mat', _LowerBound._log_dirichlet_normalization_prior, ('num_comp', 'prior_dirichlet'), 'log_dirichlet_normalization_prior') TestLogDirichletNormalization = makeTestFaithful( 'log_dirichlet_normalization.mat', _LowerBound._log_dirichlet_normalization, ('num_obs', 'num_comp', 'prior_dirichlet'), 'log_dirichlet_normalization') TestLogDirichletConst = makeTestFaithful('log_dirichlet_const.mat', _LowerBound._log_dirichlet_const, ('prior_dirichlet',), 'log_dirichlet_const') TestLogWishartConst = makeTestFaithful('log_wishart_const.mat', _LowerBound._log_wishart_const, ('num_features', 'nws_dof', 'nws_scale_matrix'), 'log_wishart_const', max_diff=1) TestExpectLogPx = makeTestFaithful('expect_log_px.mat',
import unittest from flowvb.core._ess import _ESS from tests.test_old_faithful import makeTestFaithful TestUpdateSmmMean = makeTestFaithful( "smm_mean.mat", _ESS._update_smm_mean, ("num_obs", "num_comp", "latent_scaled_resp", "latent_resp", "latent_scale"), "smm_mean", load_data=True, ) TestUpdateSmmCovar = makeTestFaithful( "smm_covar.mat", _ESS._update_smm_covar, ("num_obs", "num_features", "num_comp", "latent_resp", "latent_scale", "latent_scaled_resp", "smm_mean"), "smm_covar", load_data=True, ) TestUpdateSmmMixweights = makeTestFaithful( "smm_mixweights.mat", _ESS._update_smm_mixweights, ("num_obs", "latent_resp"), "smm_mixweights" ) if __name__ == "__main__": suite = unittest.TestLoader().loadTestsFromTestCase(TestUpdateSmmMean) suite.addTest(unittest.TestLoader().loadTestsFromTestCase(TestUpdateSmmCovar)) suite.addTest(unittest.TestLoader().loadTestsFromTestCase(TestUpdateSmmMixweights)) unittest.TextTestRunner(verbosity=2).run(suite)